Update test_block_multihead_attention.py

yinfan98 · web-flow · commit 5dd9b13ada50 · 2024-06-04T09:34:30.000+08:00
diff --git a/test/legacy_test/test_block_multihead_attention.py b/test/legacy_test/test_block_multihead_attention.py
@@ -277,7 +277,6 @@ def block_cache_to_naive_cache(
     return out_cache_k, out_cache_v
 
 
-# MHA
 @unittest.skipIf(
     not core.is_compiled_with_cuda()
     or get_cuda_version() < 11040
