Skip to content

GitLab

Commit 409431b8 authored by skrider's avatar skrider
Browse files

only test trivial block size

parent 14b190bc
Hide whitespace changes
Inline Side-by-side
Showing with 5 additions and 6 deletions
cutlass @ 751eb9a8
Compare bbe579a9...751eb9a8
Subproject commit bbe579a9e3beb6ea6626d9227ec32d0dae119a49
Subproject commit 751eb9a8859ac36bfc77551f9e4a957c31a5a8b1
tests/test_flash_attn.py
View file @ 409431b8
......@@ -1832,11 +1832,10 @@ def test_flash_attn_splitkv(
# @pytest.mark.parametrize("rotary_interleaved", [False])
@pytest.mark.parametrize("rotary_fraction", [0.0, 0.5, 1.0])
# @pytest.mark.parametrize("rotary_fraction", [0.0])
@pytest.mark.parametrize("paged_kv_block_size", [None, 256])
# @pytest.mark.parametrize("paged_kv_block_size", [256, 512])
# @pytest.mark.parametrize("paged_kv_block_size", [256])
@pytest.mark.parametrize("has_batch_idx", [False, True])
# @pytest.mark.parametrize("has_batch_idx", [False])
# @pytest.mark.parametrize("paged_kv_block_size", [None, 256, 512])
@pytest.mark.parametrize("paged_kv_block_size", [256])
# @pytest.mark.parametrize("has_batch_idx", [False, True])
@pytest.mark.parametrize("has_batch_idx", [False])
@pytest.mark.parametrize("d", [32, 59, 64, 80, 128, 256])
# @pytest.mark.parametrize("d", [32, 64, 96, 128, 160, 192, 224, 256])
# @pytest.mark.parametrize('d', [32, 40, 64, 80, 96, 128, 160, 192])
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment