upd

Author: yzh119

scripts/task_jit_run_tests_part2.sh

@@ -11,10 +11,10 @@ if [ "$SKIP_INSTALL" = "0" ]; then
 fi
 
 # Run each test file separately to isolate CUDA memory issues
-pytest -s tests/utils/test_block_sparse.py
 pytest -s tests/utils/test_jit_example.py
 pytest -s tests/utils/test_jit_warmup.py
 pytest -s tests/utils/test_norm.py
+pytest -s tests/attention/test_block_sparse.py
 pytest -s tests/attention/test_rope.py
 pytest -s tests/attention/test_mla_page.py
 pytest -s tests/utils/test_quantization.py