16 lines
343 B
Bash
Executable File
16 lines
343 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_group_gemm.py
|
|
pytest -s tests/test_logits_cap.py
|
|
pytest -s tests/test_sliding_window.py
|
|
pytest -s tests/test_tensor_cores_decode.py
|
|
pytest -s tests/test_batch_decode_kernels.py
|
|
#pytest -s tests/test_alibi.py
|