diff --git a/test/test_cuda.py b/test/test_cuda.py index b2d5fb1592f46..7a3ca272f96be 100644 --- a/test/test_cuda.py +++ b/test/test_cuda.py @@ -625,7 +625,7 @@ def _check_default(): gcn_arch = str( torch.cuda.get_device_properties(0).gcnArchName.split(":", 1)[0] ) - if gcn_arch in ["gfx90a", "gfx942", "gfx950", "gfx1200", "gfx1201"]: + if gcn_arch in ["gfx90a", "gfx942", "gfx950", "gfx1100", "gfx1101", "gfx1102", "gfx1150", "gfx1151", "gfx1200", "gfx1201"]: self.assertTrue(default == torch._C._BlasBackend.Cublaslt) else: self.assertTrue(default == torch._C._BlasBackend.Cublas)