From f4d3723a15ae794c7f802d674fe28a9a1362c714 Mon Sep 17 00:00:00 2001 From: Xiongfei Wei Date: Tue, 3 Dec 2024 22:51:54 +0000 Subject: [PATCH] enable the test --- test/test_tpu_paged_attention_kernel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_tpu_paged_attention_kernel.py b/test/test_tpu_paged_attention_kernel.py index 3fcab456844..746439ba4d0 100644 --- a/test/test_tpu_paged_attention_kernel.py +++ b/test/test_tpu_paged_attention_kernel.py @@ -112,7 +112,7 @@ def setUp(self): num_queries_per_compute_block=(16, 32), block_kv_size=(128, 256), ) - def _test_paged_attention_without_query_padding( + def test_paged_attention_without_query_padding( self, dtype, page_size,