diff --git a/test/smoke_test.py b/test/smoke_test.py index 63b35d04bed..f965c6f6aa4 100644 --- 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