From 317c3cda8fab867b608012926d66754d2d05f548 Mon Sep 17 00:00:00 2001 From: Ethan Wee Date: Thu, 14 Aug 2025 07:35:57 -0700 Subject: [PATCH 1/2] Update build_rocm.sh --- manywheel/build_rocm.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index dc84b4832..9e63c046f 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -359,7 +359,7 @@ if [ ${PYTORCH_VERSION%%\.*} -ge 2 ]; then 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)" # Use "triton" for dev builds, else "pytorch-triton-rocm" # Temp: Currently enabling for rocm7.1_internal_testing branch only but plan to expand it to other branches - if [[ "$PYTORCH_VERSION_FULL" == *"2.9.0a0"* ]]; then + if [[ "$PYTORCH_VERSION_FULL" == *"2.9.0a0"* && $ROCM_INT -gt 70000 ]]; then PKG="triton" else PKG="pytorch-triton-rocm" From 31d2ad012d2b05be30c1f0d26a243977a211c9f7 Mon Sep 17 00:00:00 2001 From: Ethan Wee Date: Thu, 14 Aug 2025 09:39:24 -0700 Subject: [PATCH 2/2] Update build_rocm.sh --- manywheel/build_rocm.sh | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh index 9e63c046f..0c9e0b819 100755 --- a/manywheel/build_rocm.sh +++ b/manywheel/build_rocm.sh @@ -359,7 +359,7 @@ if [ ${PYTORCH_VERSION%%\.*} -ge 2 ]; then 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)" # Use "triton" for dev builds, else "pytorch-triton-rocm" # Temp: Currently enabling for rocm7.1_internal_testing branch only but plan to expand it to other branches - if [[ "$PYTORCH_VERSION_FULL" == *"2.9.0a0"* && $ROCM_INT -gt 70000 ]]; then + if [[ $ROCM_INT -gt 70000 ]]; then PKG="triton" else PKG="pytorch-triton-rocm"