HappyandGitHub 9869453c42 Update test_flash_attn.py (#17102) ...

Signed-off-by: ShuaibinLi <lishuaibin@live.cn>

2025-04-26 22:17:35 +00:00

..

conftest.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_attention_selector.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_attention.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_blocksparse_attention.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_cache.py

Allocate kv_cache with stride order (#16605)

2025-04-25 22:03:31 -07:00

test_cascade_flash_attn.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_encoder_decoder_attn.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_flash_attn.py

Update test_flash_attn.py (#17102)

2025-04-26 22:17:35 +00:00

test_flashinfer.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_flashmla.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_lightning_attn.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_merge_attn_states.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_mha_attn.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_mla_decode_cpu.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_prefix_prefill.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_rocm_attention_selector.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00

test_triton_decode_attention.py

Categorize tests/kernels/ based on kernel type (#16799)

2025-04-23 09:21:07 -04:00