From abd41e5f77cc5ab018094fdf3f8279111d2de320 Mon Sep 17 00:00:00 2001 From: Scott Roy <161522778+metascroy@users.noreply.github.com> Date: Tue, 28 Jan 2025 13:34:58 -0800 Subject: [PATCH] Add torchao/experimental CI test (#1586) * add torchao/experimental CI test * up * up * up * up * up * up --- .../workflows/torchao_experimental_test.yml | 42 +++++++++++++++++++ 1 file changed, 42 insertions(+) create mode 100644 .github/workflows/torchao_experimental_test.yml diff --git a/.github/workflows/torchao_experimental_test.yml b/.github/workflows/torchao_experimental_test.yml new file mode 100644 index 0000000000..c1419bccc6 --- /dev/null +++ b/.github/workflows/torchao_experimental_test.yml @@ -0,0 +1,42 @@ +name: Run TorchAO Experimental Tests + +on: + push: + branches: + - main + - 'gh/**' + pull_request: + branches: + - main + - 'gh/**' + +jobs: + test: + strategy: + matrix: + runner: [macos-14] + runs-on: ${{matrix.runner}} + defaults: + run: + shell: bash -el {0} + steps: + - name: Checkout repo + uses: actions/checkout@v3 + with: + submodules: true + - name: Setup environment + uses: conda-incubator/setup-miniconda@v3 + with: + python-version: "3.10" + miniconda-version: "latest" + activate-environment: venv + - name: Install requirements + run: | + conda activate venv + pip install --extra-index-url "https://download.pytorch.org/whl/nightly/cpu" torch=="2.6.0.dev20250104" + pip install numpy + USE_CPP=1 pip install . + - name: Run tests + run: | + conda activate venv + python torchao/experimental/tests/test_packed_linear_int8_dynamic_activation_intx_weight_layout.py