42 lines
1.0 KiB
Bash
Executable File
42 lines
1.0 KiB
Bash
Executable File
#!/bin/bash
|
|
# Test harness: runs a test in screen, logs to kernel-test.log
|
|
# Usage: ./run_test.sh <test_file>
|
|
# Example: ./run_test.sh tests/unit/test_fmha_v3.py
|
|
|
|
set -e
|
|
|
|
TEST_FILE=$1
|
|
if [ -z "$TEST_FILE" ]; then
|
|
echo "Usage: $0 <test_file>"
|
|
exit 1
|
|
fi
|
|
|
|
# --- CLEANUP ---
|
|
if screen -list | grep -q kernel-test; then
|
|
echo "Killing existing kernel-test screen..."
|
|
screen -S kernel-test -X quit 2>/dev/null || true
|
|
sleep 2
|
|
pkill -f "python.*test_fmha" 2>/dev/null || true
|
|
sleep 1
|
|
fi
|
|
|
|
rm -f /tmp/kernel-test.log
|
|
|
|
# --- START ---
|
|
cd /root/dsv4-nvfp4-workspace/kernel
|
|
source /root/dsv4-nvfp4-workspace/venv/bin/activate
|
|
export PYTHONPATH=/root/dsv4-nvfp4-workspace/kernel
|
|
|
|
echo "Running: python -u $TEST_FILE"
|
|
echo "Log: /tmp/kernel-test.log"
|
|
screen -dmS kernel-test bash -c 'python -u '"$TEST_FILE"' > /tmp/kernel-test.log 2>&1'
|
|
sleep 2
|
|
|
|
if screen -list | grep -q kernel-test; then
|
|
echo "OK: screen kernel-test is running"
|
|
else
|
|
echo "FAIL: screen did not start. Log below:"
|
|
cat /tmp/kernel-test.log 2>/dev/null
|
|
exit 1
|
|
fi
|