Merge branch 'a100' into lm_workload_tf32

priyakasimbeg · priyakasimbeg · commit b6f37ac0614e · 2025-12-18T19:08:33.000Z
diff --git a/algoperf/pytorch_utils.py b/algoperf/pytorch_utils.py
@@ -20,6 +20,7 @@
 
 
 def pytorch_setup() -> Tuple[bool, int, torch.device, int]:
+  torch.set_float32_matmul_precision('high')
   # PyTorch set TF32
   # torch.backends.fp32_precision = "ieee"
   # torch.backends.cuda.matmul.fp32_precision = "tf32"
