Skip to content

MX Updated to_blocked to not call nn.pad (#1762) #904

MX Updated to_blocked to not call nn.pad (#1762)

MX Updated to_blocked to not call nn.pad (#1762) #904

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