Skip to content

Parallelize and optimize tests #1184

Parallelize and optimize tests

Parallelize and optimize tests #1184

name: Benchmarks
on:
push:
branches: [ master ]
pull_request:
jobs:
run_benchmarks:
runs-on: ubuntu-latest
timeout-minutes: 10
steps:
- uses: actions/checkout@v3
- name: Set up Python
uses: actions/setup-python@v3
with:
python-version: 3.9
- name: Cache dependencies
uses: actions/cache@v3
with:
path: ~/.cache/pip
key: Key-v1-3.9-${{ hashFiles('requirements.txt') }}-${{ hashFiles('requirements-dev.txt') }}
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install -r requirements-dev.txt
- name: Build bitsandbytes
run: |
pip install bitsandbytes==0.41.1
- name: Build hivemind
run: |
pip install .
- name: Benchmark
run: |
cd benchmarks
python benchmark_throughput.py --preset minimalistic
python benchmark_tensor_compression.py
python benchmark_dht.py