Assert Runtime Error for torch.compile on Python 3.11 (#7387)

5d9bc5e5 · Omkar Salpekar · GitHub · 81f0c0c9 · 5d9bc5e5
Unverified Commit 5d9bc5e5 authored Mar 03, 2023 by Omkar Salpekar Committed by GitHub Mar 03, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

test/smoke_test.py test/smoke_test.py +2 -0

No files found.
--- a/test/smoke_test.py
+++ b/test/smoke_test.py
@@ -39,6 +39,8 @@ def smoke_test_compile() -> None:
    except RuntimeError:
        if platform == "win32":
            print("Successfully caught torch.compile RuntimeError on win")
+        elif sys.version_info >= (3, 11, 0):
+            print("Successfully caught torch.compile RuntimeError on Python 3.11")
        else:
            raise