[EZ] Test torch.compile on 3.12 (#1871)

Test plan: CI
pytorch · Jun 17, 2024 · 2681a99 · 2681a99
1 parent 3b032fc
commit 2681a99
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/smoke_test/smoke_test.py b/test/smoke_test/smoke_test.py
@@ -162,8 +162,8 @@ def smoke_test_cuda(package: str, runtime_error_check: str, torch_compile_check:
                 version = imported_module._extension._check_cuda_version()
             print(f"{module['name']} CUDA: {version}")
 
-     # torch.compile is available on macos-arm64 and Linux for python 3.8-3.11
-    if torch_compile_check == "enabled" and sys.version_info < (3, 12, 0) and (
+     # torch.compile is available on macos-arm64 and Linux for python 3.8-3.12
+    if torch_compile_check == "enabled" and sys.version_info < (3, 13, 0) and (
         (target_os == "linux" and torch.cuda.is_available()) or
         target_os == "macos-arm64"):
         smoke_test_compile()