Skip to content

GitLab

"git@developer.sourcefind.cn:renzhc/diffusers_dcu.git" did not exist on "c36f8487df35895421c15f351c7d360bd6800e5c"
Unverified Commit 5d9bc5e5 authored by Omkar Salpekar's avatar Omkar Salpekar Committed by GitHub
Browse files

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

parent 81f0c0c9
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
test/smoke_test.py
View file @ 5d9bc5e5
...@@ -39,6 +39,8 @@ def smoke_test_compile() -> None: ...@@ -39,6 +39,8 @@ def smoke_test_compile() -> None:
except RuntimeError: except RuntimeError:
if platform == "win32": if platform == "win32":
print("Successfully caught torch.compile RuntimeError on win") 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: else:
raise raise
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment