Skip to content

Commit 06e5126

Browse files
committed
Revert "[main] Conditionalize triton package name on rocm version (#91)"
This reverts commit 66b8d8d.
1 parent 2aba8be commit 06e5126

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

manywheel/build_rocm.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -359,7 +359,7 @@ if [ ${PYTORCH_VERSION%%\.*} -ge 2 ]; then
359359
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)"
360360
# Use "triton" for dev builds, else "pytorch-triton-rocm"
361361
# 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
362+
if [[ "$PYTORCH_VERSION_FULL" == *"2.9.0a0"* ]]; then
363363
PKG="triton"
364364
else
365365
PKG="pytorch-triton-rocm"

0 commit comments

Comments
 (0)