AssertionError -> AttributeError in cuBLASModule (#112606)

jansel · pytorchmergebot · commit c1e2ccdb97f1 · 2023-11-01T23:23:10.000Z
Pull Request resolved: #112606 Approved by: https://github.com/eellison
diff --git a/torch/backends/cuda/__init__.py b/torch/backends/cuda/__init__.py
@@ -121,7 +121,7 @@ def __getattr__(self, name):
             return torch._C._get_cublas_allow_fp16_reduced_precision_reduction()
         elif name == "allow_bf16_reduced_precision_reduction":
             return torch._C._get_cublas_allow_bf16_reduced_precision_reduction()
-        raise AssertionError("Unknown attribute " + name)
+        raise AttributeError("Unknown attribute " + name)
 
     def __setattr__(self, name, value):
         if name == "allow_tf32":
@@ -130,7 +130,7 @@ def __setattr__(self, name, value):
             return torch._C._set_cublas_allow_fp16_reduced_precision_reduction(value)
         elif name == "allow_bf16_reduced_precision_reduction":
             return torch._C._set_cublas_allow_bf16_reduced_precision_reduction(value)
-        raise AssertionError("Unknown attribute " + name)
+        raise AttributeError("Unknown attribute " + name)
 
 
 _LinalgBackends = {
