Skip to content

Add CUTLASS-based W4A4 #101

Add CUTLASS-based W4A4

Add CUTLASS-based W4A4 #101

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