Skip to content

Commit e4df565

Browse files
authored
Merge branch 'release/2.8' into fix_torch_macros_for_miopen
2 parents 8ade7b5 + 8d1791e commit e4df565

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed
Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
5e5685356b9fc7b5ad9cdf4e510a1994a5b8601a
1+
f9e5bf54a2fe1a6262a41b27b38180cdb6fae6a2

.github/scripts/build_triton_wheel.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -101,8 +101,12 @@ def build_triton(
101101

102102
triton_repo = "https://github.com/openai/triton"
103103
if device == "rocm":
104-
triton_pkg_name = "pytorch-triton-rocm"
105104
triton_repo = "https://github.com/ROCm/triton"
105+
rocm_version = get_rocm_version() # e.g., "7.0.1"
106+
if tuple(map(int, rocm_version.split("."))) > (7, 0, 0):
107+
triton_pkg_name = "triton"
108+
else:
109+
triton_pkg_name = "pytorch-triton-rocm"
106110
elif device == "xpu":
107111
triton_pkg_name = "pytorch-triton-xpu"
108112
triton_repo = "https://github.com/intel/intel-xpu-backend-for-triton"

0 commit comments

Comments
 (0)