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

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

Differential Revision: D47803991

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

diff --git a/dev-requirements.txt b/dev-requirements.txt
index 9d82cac567..30463a7562 100644
--- 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
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