From 6dcffc6f43f3e13ef644836e6bd9a5c18acbc774 Mon Sep 17 00:00:00 2001
From: Jason Senthil <jsenthil@meta.com>
Date: Wed, 26 Jul 2023 11:07:10 -0700
Subject: [PATCH] fix torch compile tests in OSS (#479)

Summary:
Pull Request resolved: https://github.com/pytorch/tnt/pull/479

seems module.compile is only in pytorch nightlies, so must add that to dev-requirements.txt

Differential Revision: D47803991

fbshipit-source-id: 5ae392384902c9b2286ce4c8d7020bccb8e8f78b
---
 dev-requirements.txt              | 4 ++++
 tests/framework/test_auto_unit.py | 1 -
 2 files changed, 4 insertions(+), 1 deletion(-)

diff --git a/dev-requirements.txt b/dev-requirements.txt
index 9d82cac567..c200cd120c 100644
--- 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
index 4b43fc07be..7d911a440a 100644
--- 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