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

c1e2095c · Andrey Talman · GitHub · 1fda0e85 · c1e2095c
Unverified Commit c1e2095c authored Dec 01, 2023 by Andrey Talman Committed by GitHub Dec 01, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 1 deletion

test/smoke_test.py test/smoke_test.py +4 -1

No files found.
--- 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")