use new tf32 APIs for torch 2.9+ (#3467) [skip ci]

* use new tf32 APIs for torch 2.9+ * also upgrade cce for tf32 fixes and lint
2026-03-06 11:40:32 -05:00
parent c119382337
commit fc2d63ee5f
6 changed files with 18 additions and 7 deletions
--- a/tests/test_tensor_parallel_batch_size.py
+++ b/tests/test_tensor_parallel_batch_size.py
@@ -4,6 +4,7 @@ from unittest.mock import patch

 import addict
 import pytest
+
 from axolotl.utils.config import normalize_config, validate_config
 from axolotl.utils.dict import DictDefault