up

pytorch · Jan 28, 2025 · 04f3f03 · 04f3f03
1 parent 95525a4
commit 04f3f03
Showing 1 changed file with 7 additions and 6 deletions.
diff --git a/.github/workflows/torchao_experimental_test.yml b/.github/workflows/torchao_experimental_test.yml
@@ -21,13 +21,14 @@ jobs:
         uses: actions/checkout@v3
         with:
           submodules: true
-      - name: Setup Python
-        uses: actions/setup-python@v2
+      - uses: conda-incubator/setup-miniconda@v3
         with:
-          python-version: 3.10.11
-      - name: Install TorchAO
-        run: |
-          USE_CPP=1 pip install .
+          python-version: "3.10"
+          miniconda-version: "latest"
+          activate-environment: venv
       - name: Run tests
         run: |
+          pip install --extra-index-url "https://download.pytorch.org/whl/nightly/cpu" torch=="2.6.0.dev20250104"
+          USE_CPP=1 pip install .
+
           python torchao/experimental/tests/test_packed_linear_int8_dynamic_activation_intx_weight_layout.py