File size: 984 Bytes
b88c26d | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 | #!/bin/bash
# BURT-IMMA Test Runner
# License: BSL-1.1
# Contact: jessica@collectivekitty.com
set -e
echo "============================================"
echo " BURT-IMMA Test Suite"
echo "============================================"
RUN_CUDA=0
for arg in "$@"; do
case $arg in
--cuda)
RUN_CUDA=1
shift
;;
esac
done
echo ""
echo "[1/2] Running Python test suite..."
pytest tests/ -v --tb=short
if [ $RUN_CUDA -eq 1 ]; then
echo ""
echo "[2/2] Running CUDA kernel tests..."
if [ -f build/test_kernels ]; then
./build/test_kernels
else
echo " CUDA test binary not found. Run ./scripts/build.sh first."
exit 1
fi
else
echo ""
echo "[2/2] Skipping CUDA kernel tests (pass --cuda to enable)"
fi
echo ""
echo "============================================"
echo " All tests passed."
echo "============================================"
|