Pass environement variables for caching torch compiles

Author: faymanns

.github/workflows/test_and_deploy.yml

@@ -45,9 +45,10 @@ jobs:
 
       # this runs the platform-specific tests declared in tox.ini
       - name: Test with tox
-        run: python -m tox
         env:
           PLATFORM: ${{ matrix.platform }}
+          TORCHINDUCTOR_CACHE_DIR: ${{ runner.temp }}/torchinductor
+        run: python -m tox
 
       - name: Upload coverage reports to Codecov
         uses: codecov/codecov-action@v4

tox.ini

@@ -24,6 +24,10 @@ platform =
 passenv =
     CI
     GITHUB_ACTIONS
+    USERNAME
+    USER
+    LOGNAME
+    TORCHINDUCTOR_CACHE_DIR
 extras =
     test,examples
 commands =