fix torch compile tests in OSS (pytorch#479)

Summary: Pull Request resolved: pytorch#479 seems module.compile is only in pytorch nightlies, so must add that to dev-requirements.txt Differential Revision: D47803991 fbshipit-source-id: 5ae392384902c9b2286ce4c8d7020bccb8e8f78b
JKSenthil · Jul 26, 2023 · 6dcffc6 · 6dcffc6
1 parent fe9ea1c
commit 6dcffc6
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -1,5 +1,9 @@
 parameterized
 pytest
+
+--pre --index-url https://download.pytorch.org/whl/nightly/cpu
+torch
+
 pytest-cov
 torcheval-nightly
 torchsnapshot-nightly

diff --git a/tests/framework/test_auto_unit.py b/tests/framework/test_auto_unit.py
@@ -16,7 +16,6 @@
 COMPILE_AVAIL = False
 if is_torch_version_geq_1_13():
     COMPILE_AVAIL = True
-    import torch._dynamo
 
 from torch.distributed import GradBucket, launcher
 from torch.distributed.fsdp import FullyShardedDataParallel as FSDP