Skip to content

Commit 2360bd1

Browse files
committed
yapf
Signed-off-by: Yu Chin Fabian Lim <flim@sg.ibm.com>
1 parent 83565de commit 2360bd1

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/kernels/mamba/test_mamba_ssm_ssd.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -269,7 +269,7 @@ def test_mamba_chunk_scan_cont_batch(d_head, n_heads, seq_len_chunk_size_cases,
269269
# (i.e. chunked prefill)
270270

271271
seqlen, chunk_size, num_examples, cases = seq_len_chunk_size_cases
272-
272+
273273
# TODO: the irregular chunk size cases have some issues and require higher
274274
# tolerance. This is to be invesigated
275275
if chunk_size not in {8, 256}:

0 commit comments

Comments
 (0)