Correct torchao check for TensorCoreTiledLayout (#1886)

pytorch · Oct 23, 2024 · 17ba37d · 17ba37d
1 parent 1e5f0d5
commit 17ba37d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchtune/utils/_import_guard.py b/torchtune/utils/_import_guard.py
@@ -20,7 +20,7 @@
 _USE_NEW_TENSOR_CORE_TILED_LAYOUT_API = _is_fbcode() or (
     not _is_fbcode()
     and (
-        ("dev" not in torchao_version and torchao_version >= "0.6.0")
+        ("dev" not in torchao_version and torchao_version >= "0.7.0")
         or (
             "dev" in torchao_version
             and _nightly_version_ge(torchao_version, "2024-10-10")