fix torch compile tests in OSS (pytorch#479)

Summary: Pull Request resolved: pytorch#479 Differential Revision: D47803991 fbshipit-source-id: 936860f62e484aefd7959531ecf77536dd67273c
JKSenthil · Jul 26, 2023 · 1ea9394 · 1ea9394
1 parent fe9ea1c
commit 1ea9394
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 1 deletion.
diff --git a/dev-requirements.txt b/dev-requirements.txt
@@ -1,5 +1,10 @@
 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