automatically enable tf32 if supported (#3473) [skip ci]

* automatically enable tf32 if supported * update fixtures * handle only when True * Address CR comments * address readability from pr comment * simplify
2026-03-16 23:47:00 -04:00
parent d230cbbde3
commit 830e9f7eaf
8 changed files with 29 additions and 8 deletions
--- a/tests/test_validation_dataset.py
+++ b/tests/test_validation_dataset.py
@@ -68,6 +68,7 @@ class TestValidationCheckDatasetConfig(BaseValidation):
            cfg,
            capabilities={
                "bf16": "false",
+                "tf32": "false",
                "n_gpu": 1,
                "compute_capability": "8.0",
            },