Skip to content

Commit 867521e

Browse files
authored
Fix torch.compile smoke test include it for py 3.12, 3.13 (#8876)
1 parent 4315f8a commit 867521e

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

test/smoke_test.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22

33
import os
44
import sys
5+
import sysconfig
56
from pathlib import Path
67

78
import torch
@@ -134,8 +135,8 @@ def main() -> None:
134135
smoke_test_torchvision_decode_jpeg("cuda")
135136
smoke_test_torchvision_resnet50_classify("cuda")
136137

137-
# TODO: remove once pytorch/pytorch#110436 is resolved
138-
if sys.version_info < (3, 12, 0):
138+
# torch.compile is not supported on Python 3.14+ and Python built with GIL disabled
139+
if sys.version_info < (3, 14, 0) and not sysconfig.get_config_var("Py_GIL_DISABLED"):
139140
smoke_test_compile()
140141

141142
if torch.backends.mps.is_available():

0 commit comments

Comments
 (0)