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 eab82c5 commit 17731a9Copy full SHA for 17731a9
manywheel/build_rocm.sh
@@ -358,7 +358,7 @@ if [ ${PYTORCH_VERSION%%\.*} -ge 2 ]; then
358
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)"
359
# Use "triton" for dev builds, else "pytorch-triton-rocm"
360
# Temp: Currently enabling for rocm7.1_internal_testing branch only but plan to expand it to other branches
361
- if [[ $ROCM_INT -gt 70000 ]]; then
+ if [[ "$PYTORCH_VERSION_FULL" == *"2.9.0a0"* ]]; then
362
PKG="triton"
363
else
364
PKG="pytorch-triton-rocm"
0 commit comments