We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2aba8be commit 06e5126Copy full SHA for 06e5126
manywheel/build_rocm.sh
@@ -359,7 +359,7 @@ if [ ${PYTORCH_VERSION%%\.*} -ge 2 ]; then
359
TRITON_CONSTRAINT="platform_system == 'Linux' and platform_machine == 'x86_64'$(if [[ $(ver "$PYTORCH_VERSION") -le $(ver "2.5") ]]; then echo " and python_version < '3.13'"; fi)"
360
# Use "triton" for dev builds, else "pytorch-triton-rocm"
361
# Temp: Currently enabling for rocm7.1_internal_testing branch only but plan to expand it to other branches
362
- if [[ $ROCM_INT -gt 70000 ]]; then
+ if [[ "$PYTORCH_VERSION_FULL" == *"2.9.0a0"* ]]; then
363
PKG="triton"
364
else
365
PKG="pytorch-triton-rocm"
0 commit comments