Add torchao/experimental CI test (#1586)

* add torchao/experimental CI test * up * up * up * up * up * up
pytorch · Jan 28, 2025 · abd41e5 · abd41e5
1 parent e151d6a
commit abd41e5
Showing 1 changed file with 42 additions and 0 deletions.
diff --git a/.github/workflows/torchao_experimental_test.yml 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