8000 add unit test · pytorch/ao@1f2b2e1 · GitHub
[go: up one dir, main page]
More Web Proxy on the site http://driver.im/
Skip to content
Run TorchAO Experimental Tests 8000

[draft] add all_gather_into_tensor #5301

Sign in to view logs

[draft] add all_gather_into_tensor

[draft] add all_gather_into_tensor #5301

Workflow file for this run

.github/workflows/torchao_experimental_test.yml at 1f2b2e1
6A06
name: Run TorchAO Experimental Tests
on:
push:
branches:
- main
- 'gh/**'
pull_request:
branches:
- main
- 'gh/**'
jobs:
test-cpu-ops:
strategy:
matrix:
runner: [macos-14, linux.arm64.2xlarge]
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 mac
if: runner.os == 'macOS'
run: |
conda activate venv
# Install executorch first because it installs its own version
# of torch and torchao, which we do not want to use
pip install executorch
pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu --force-reinstall
pip install -r dev-requirements.txt
USE_CPP=1 TORCHAO_BUILD_KLEIDIAI=1 pip install .
- name: Install requirements linux
if: runner.os == 'Linux'
run: |
conda activate venv
pip install torch==2.7.0 --index-url https://download.pytorch.org/whl/cpu --force-reinstall
pip install -r dev-requirements.txt
BUILD_TORCHAO_EXPERIMENTAL=1 TORCHAO_BUILD_CPU_AARCH64=1 TORCHAO_BUILD_KLEIDIAI=1 TORCHAO_ENABLE_ARM_NEON_DOT=1 TORCHAO_PARALLEL_BACKEND=OPENMP 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
python torchao/experimental/tests/test_quant_passes.py
- name: Run kernels/cpu/aarch64/tests
if: runner.os == 'macOS'
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
if: runner.os == 'macOS'
run: |
conda activate venv
pushd torchao/experimental/ops/tests
sh build_and_run_tests.sh
rm -rf /tmp/cmake-out
popd
- name: ET ops build
if: runner.os == 'macOS'
run: |
conda activate venv
pushd torchao/experimental
sh build_torchao_ops.sh executorch
popd
test-mps-ops:
strategy:
matrix:
runner: [macos-m1-stable]
runs-on: ${{matrix.runner}}
steps:
- name: Print machine info
run: |
uname -a
if [ $(uname -s) == Darwin ]; then
sysctl machdep.cpu.brand_string
sysctl machdep.cpu.core_count
fi
- name: Checkout repo
uses: actions/checkout@v3
with:
submodules: true
- name: Create conda env
run: |
conda create -yn test-mps-ops-env python=3.11
- name: Activate conda env
run: |
source activate base
conda activate test-mps-ops-env
- name: Install torch
run: |
conda run -n test-mps-ops-env pip install torch --index-url "https://download.pytorch.org/whl/nightly/cpu"
- name: Print torch version
run: |
conda run -n test-mps-ops-env python -c "import torch; print(torch.__version__)"
- name: Install requirements
run: |
source activate base
conda activate test-mps-ops-env
pip install -r dev-requirements.txt
pip install pyyaml importlib-metadata
- name: Print pip freeze
run: |
conda run -n test-mps-ops-env pip freeze
- name: Print current directory
run: |
conda run -n test-mps-ops-env python -c "import os; print(os.getcwd())"
- name: Build ao with experimental mps ops
run: |
source activate base
conda activate test-mps-ops-env
USE_CPP=1 TORCHAO_BUILD_EXPERIMENTAL_MPS=1 pip install .
- name: Run mps tests
run: |
pushd torchao/experimental/ops/mps/test
conda run -n test-mps-ops-env python test_lowbit.py
conda run -n test-mps-ops-env python test_quantizer.py
popd
0