Skip to content

Commit

Permalink
[fbsync] Revert "Update smoke_test.py - temporary disable torch.compi…
Browse files Browse the repository at this point in the history
…le tests" (#8765)

Reviewed By: vmoens

Differential Revision: D68021987

fbshipit-source-id: 6c01ff26ad5935d4facc2d93cff181d6677aa0d2
  • Loading branch information
NicolasHug authored and facebook-github-bot committed Jan 13, 2025
1 parent dd40041 commit 7f4e6c2
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions test/smoke_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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")
Expand Down

0 comments on commit 7f4e6c2

Please sign in to comment.