From 7f4e6c2130d6ca8966d069454e8526f15e14a9d1 Mon Sep 17 00:00:00 2001 From: Nicolas Hug Date: Mon, 13 Jan 2025 03:58:41 -0800 Subject: [PATCH] [fbsync] Revert "Update smoke_test.py - temporary disable torch.compile tests" (#8765) Reviewed By: vmoens Differential Revision: D68021987 fbshipit-source-id: 6c01ff26ad5935d4facc2d93cff181d6677aa0d2 --- test/smoke_test.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/test/smoke_test.py b/test/smoke_test.py index 647e2f45c8f..3a44ae3efe9 100644 --- a/test/smoke_test.py +++ b/test/smoke_test.py @@ -102,9 +102,8 @@ def main() -> None: smoke_test_torchvision_resnet50_classify("cuda") # TODO: remove once pytorch/pytorch#110436 is resolved - # Temporary Disabling compile test. Untill triton with Manylinux2014 is available - # if sys.version_info < (3, 12, 0): - # smoke_test_compile() + if sys.version_info < (3, 12, 0): + smoke_test_compile() if torch.backends.mps.is_available(): smoke_test_torchvision_resnet50_classify("mps")