sglang_v0.5.2/flashinfer_0.3.1/scripts/run_test_blackwell_gemm_ker...

10 lines
221 B
Bash

#!/bin/bash
set -eo pipefail
set -x
pytest -s tests/test_mm_fp4.py
pytest -s tests/test_groupwise_scaled_gemm_fp8.py
pytest -s tests/test_groupwise_scaled_gemm_mxfp4.py
pytest -s tests/test_cute_dsl_blockscaled_gemm.py