Run linter again

pytorch · Apr 22, 2024 · c6040cf · c6040cf
1 parent 312bef1
commit c6040cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_pallas.py b/test/test_pallas.py
@@ -40,7 +40,7 @@ def _pagedattention_generate_qkv(
       head_dim,
       dtype=torch.float32,
   ):
-    # assert max_seq_len % page_size == 0
+    assert max_seq_len % page_size == 0
     pages_per_sequence = max_seq_len // page_size
     batch_size = len(seq_lens)
     total_pages = batch_size * pages_per_sequence