diff --git a/manywheel/build_rocm.sh b/manywheel/build_rocm.sh
index dc84b4832..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"* ]]; then
+        if [[ $ROCM_INT -gt 70000 ]]; then
             PKG="triton"
         else
             PKG="pytorch-triton-rocm"