diff --git a/.github/workflows/torchao_experimental_test.yml b/.github/workflows/torchao_experimental_test.yml index a87af3648e..1c9b362562 100644 --- 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