fix torch compile tests in OSS

.github/workflows/build_docs.yaml

@@ -26,7 +26,7 @@ jobs:
           set -eux
           conda activate test
           conda install pytorch cpuonly -c pytorch-nightly
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           pip install -r dev-requirements.txt
           python setup.py sdist bdist_wheel
           pip install dist/*.whl
@@ -36,7 +36,7 @@ jobs:
           set -eux
           conda activate test
           cd docs
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           make html
           touch build/html/.nojekyll
           cd ..

.github/workflows/nightly_build_cpu.yaml

@@ -32,7 +32,7 @@ jobs:
           set -eux
           conda activate test
           conda install pytorch cpuonly -c pytorch-nightly
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           python setup.py sdist bdist_wheel
           pip install dist/*.whl
           pip install -r dev-requirements.txt
@@ -61,7 +61,7 @@ jobs:
           set -eux
           conda activate test
           conda install pytorch cpuonly -c pytorch-nightly
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           pip install --no-build-isolation -e ".[dev]"
       - name: Upload to PyPI
         shell: bash -l {0}

.github/workflows/release_build.yaml

@@ -24,7 +24,7 @@ jobs:
           set -eux
           conda activate test
           conda install pytorch cpuonly -c pytorch-nightly
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           python setup.py sdist bdist_wheel
           pip install dist/*.whl
           pip install -r dev-requirements.txt
@@ -53,7 +53,7 @@ jobs:
           set -eux
           conda activate test
           conda install pytorch cpuonly -c pytorch-nightly
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           pip install  --no-build-isolation -e ".[dev]"
       - name: Upload to PyPI
         shell: bash -l {0}
@@ -82,7 +82,7 @@ jobs:
           set -eux
           conda activate test
           conda install pytorch cpuonly -c pytorch-nightly
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           pip install -r dev-requirements.txt
           python setup.py sdist bdist_wheel
           pip install dist/*.whl
@@ -92,7 +92,7 @@ jobs:
           set -eux
           conda activate test
           cd docs
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           RELEASE_BUILD=1 make html
           touch build/html/.nojekyll
           cd ..

.github/workflows/test.yaml

@@ -26,7 +26,7 @@ jobs:
           set -eux
           conda activate test
           conda install pytorch cpuonly -c pytorch-nightly
-          pip install -r requirements.txt
+          pip install $(grep -v '^ *#\|^torch' requirements.txt | grep .)
           pip install -r dev-requirements.txt
           pip install  --no-build-isolation -e .
       - name: Run unit tests with coverage

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