8000 [ROCm] moved gfx1100 back to experimental status for AOTriton (#166397) · pytorch/pytorch@45c3f02 · GitHub
[go: up one dir, main page]
Skip to content

Commit 45c3f02

Browse files
k-artempytorchmergebot
k-artem
authored and
pytorchmergebot
committed
[ROCm] moved gfx1100 back to experimental status for AOTriton (#166397)
According to next commit to AOTriton: ROCm/aotriton@8625c4f These changes missed in 0.11b release: #161754 Pull Request resolved: #166397 Approved by: https://github.com/jeffdaily
1 parent f5543e3 commit 45c3f02

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

‎torch/testing/_internal/common_cuda.py‎

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -60,19 +60,19 @@ def CDNA2OrLater():
6060

6161
def evaluate_platform_supports_flash_attention():
6262
if TEST_WITH_ROCM:
63-
arch_list = ["gfx90a", "gfx942", "gfx1100", "gfx1201", "gfx950"]
63+
arch_list = ["gfx90a", "gfx942", "gfx1201", "gfx950"]
6464
if os.environ.get("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "0") != "0":
65-
arch_list += ["gfx1101", "gfx1150", "gfx1151", "gfx1200"]
65+
arch_list += ["gfx1100", "gfx1101", "gfx1102", "gfx1150", "gfx1151", "gfx1200"]
6666
return evaluate_gfx_arch_within(arch_list)
6767
if TEST_CUDA:
6868
return not IS_WINDOWS and SM80OrLater
6969
return False
7070

7171
def evaluate_platform_supports_efficient_attention():
7272
if TEST_WITH_ROCM:
73-
arch_list = ["gfx90a", "gfx942", "gfx1100", "gfx1201", "gfx950"]
73+
arch_list = ["gfx90a", "gfx942", "gfx1201", "gfx950"]
7474
if os.environ.get("TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL", "0") != "0":
75-
arch_list += ["gfx1101", "gfx1150", "gfx1151", "gfx1200"]
75+
arch_list += ["gfx1100", "gfx1101", "gfx1102", "gfx1150", "gfx1151", "gfx1200"]
7676
return evaluate_gfx_arch_within(arch_list)
7777
if TEST_CUDA:
7878
return True

0 commit comments

Comments
 (0)
0