torchao_experimental_test.yml

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
          pip install pytest
          USE_CPP=1 pip install .
      - name: Run python tests
        run: |
          conda activate venv
          pytest torchao/experimental/tests/test_int8_dynamic_activation_intx_weight.py
          python torchao/experimental/tests/test_embedding_xbit_quantizer.py
      - name: Run kernels/cpu/aarch64/tests
        run: |
          conda activate venv
          pushd torchao/experimental/kernels/cpu/aarch64/tests
          sh build_and_run_tests.sh
          rm -rf /tmp/cmake-out
          popd
      - name: Run torchao/experimental/ops/tests
        run: |
          conda activate venv
          pushd torchao/experimental/ops/tests
          sh build_and_run_tests.sh
          rm -rf /tmp/cmake-out
          popd