enable the test

pytorch · Dec 3, 2024 · f4d3723 · f4d3723
1 parent 0106b2f
commit f4d3723
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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,