[Python 3.12] Fix for, Dynamo is not supported on Python 3.12+ error (#…

…8137)
pytorch · Dec 1, 2023 · c1e2095 · c1e2095
1 parent 1fda0e8
commit c1e2095
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/test/smoke_test.py b/test/smoke_test.py
@@ -90,7 +90,10 @@ def main() -> None:
     if torch.cuda.is_available():
         smoke_test_torchvision_decode_jpeg("cuda")
         smoke_test_torchvision_resnet50_classify("cuda")
-        smoke_test_compile()
+
+        # TODO: remove once pytorch/pytorch#110436 is resolved
+        if sys.version_info < (3, 12, 0):
+            smoke_test_compile()
 
     if torch.backends.mps.is_available():
         smoke_test_torchvision_resnet50_classify("mps")