Dtype assignment without trycatch

microsoft · Aug 21, 2024 · 39bc878 · 39bc878
1 parent 4c8cb3c
commit 39bc878
Showing 1 changed file with 1 addition and 8 deletions.
diff --git a/tests/unit/ops/transformer/inference/inference_test_utils.py b/tests/unit/ops/transformer/inference/inference_test_utils.py
@@ -26,14 +26,7 @@ def get_tolerances():
 def get_dtypes():
     global DTYPES
     if DTYPES is None:
-        DTYPES = [torch.float32]
-        try:
-            if get_accelerator().is_fp16_supported():
-                DTYPES.append(torch.float16)
-            if get_accelerator().is_bf16_supported():
-                DTYPES.append(torch.bfloat16)
-        except (AssertionError, AttributeError):
-            pass
+        DTYPES = get_accelerator().supported_dtypes()
     return DTYPES