update test list. rename file

Author: weireweire

scripts/run_test_blackwell_attention_kernels.sh

@@ -7,8 +7,8 @@ pytest -s tests/test_blackwell_fmha.py
 pytest -s tests/test_deepseek_mla.py
 
 # trtllm-gen
-pytest -s tests/test_trtllm_gen_context.py
-pytest -s tests/test_trtllm_gen_decode.py
+pytest -s tests/test_trtllm_gen_attention.py
+pytest -s tests/test_trtllm_gen_mla.py
 
 # cudnn
 pytest -s tests/test_cudnn_decode.py

tests/test_trtllm_gen_attention.py

@@ -472,7 +472,7 @@ def test_trtllm_batch_decode(
     sm_scale = float(1.0 / (head_dim**0.5))
 
     output = flashinfer.decode.trtllm_batch_decode_with_kv_cache(
-        q,
+        q.contiguous(),
         kv_cache,
         workspace_buffer,
         page_table,