17 lines
343 B
Bash
17 lines
343 B
Bash
#!/bin/bash
|
|
|
|
set -eo pipefail
|
|
set -x
|
|
|
|
pytest -s tests/test_blackwell_fmha.py
|
|
pytest -s tests/test_deepseek_mla.py
|
|
|
|
# trtllm-gen
|
|
pytest -s tests/test_trtllm_gen_attention.py
|
|
pytest -s tests/test_trtllm_gen_mla.py
|
|
|
|
# cudnn
|
|
pytest -s tests/test_cudnn_decode.py
|
|
pytest -s tests/test_cudnn_prefill.py
|
|
pytest -s tests/test_cudnn_prefill_deepseek.py
|