17 lines
348 B
Bash
Executable File
17 lines
348 B
Bash
Executable File
#!/bin/bash
|
|
|
|
set -eo pipefail
|
|
set -x
|
|
: ${MAX_JOBS:=$(nproc)}
|
|
: ${CUDA_VISIBLE_DEVICES:=0}
|
|
|
|
pip install -e . -v
|
|
|
|
pytest -s tests/test_block_sparse.py
|
|
pytest -s tests/test_jit_example.py
|
|
pytest -s tests/test_jit_warmup.py
|
|
pytest -s tests/test_norm.py
|
|
pytest -s tests/test_rope.py
|
|
pytest -s tests/test_mla_page.py
|
|
pytest -s tests/test_quantization.py
|