diff --git a/test/test_cuda.py b/test/test_cuda.py index d15c756fa8992..e4d7182370daa 100644 --- a/test/test_cuda.py +++ b/test/test_cuda.py @@ -840,6 +840,8 @@ def _check_default(): archs.append("gfx950") if ROCM_VERSION >= (7, 13): archs.extend(["gfx1100", "gfx1101", "gfx1151"]) + if ROCM_VERSION >= (7, 14): + archs.extend(["gfx1250"]) gcn_arch_name = torch.cuda.get_device_properties(0).gcnArchName hipblaslt_preferred = any(arch in gcn_arch_name for arch in archs) if hipblaslt_preferred: