Skip to content

Commit 8d1a031

Browse files
authored
[release/2.7] Added missing MI350_ARCH constant value (#2519)
Fixes test/distributed/tensor/test_matrix_ops.py by defining missing constant value
1 parent 56b79e1 commit 8d1a031

File tree

1 file changed

+1
-0
lines changed

1 file changed

+1
-0
lines changed

torch/testing/_internal/common_utils.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -104,6 +104,7 @@
104104

105105

106106
MI300_ARCH = ("gfx940", "gfx941", "gfx942")
107+
MI350_ARCH = ("gfx950")
107108
NAVI_ARCH = ("gfx1030", "gfx1100", "gfx1101", "gfx1200", "gfx1201")
108109
NAVI3_ARCH = ("gfx1100", "gfx1101")
109110
NAVI4_ARCH = ("gfx1200", "gfx1201")

0 commit comments

Comments
 (0)