Skip to content

Commit

Permalink
fix torch compile tests in OSS
Browse files Browse the repository at this point in the history
Differential Revision: D47803991

fbshipit-source-id: ffd45a41835a11b5e58960f377a9fa33fb6fa3e6
  • Loading branch information
JKSenthil authored and facebook-github-bot committed Jul 26, 2023
1 parent fe9ea1c commit b475e9e
Show file tree
Hide file tree
Showing 2 changed files with 5 additions and 1 deletion.
5 changes: 5 additions & 0 deletions dev-requirements.txt
Original file line number Diff line number Diff line change
@@ -1,5 +1,10 @@
parameterized
pytest

--pre
--index-url https://download.pytorch.org/whl/nightly/cpu
torch

pytest-cov
torcheval-nightly
torchsnapshot-nightly
Expand Down
1 change: 0 additions & 1 deletion tests/framework/test_auto_unit.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit b475e9e

Please sign in to comment.