Shu Wang 9e96f56efb Allocate kv_cache with stride order (#16605) ...

Signed-off-by: shuw <shuw@nvidia.com>

2025-04-25 22:03:31 -07: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

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

2025-04-23 09:21:07 -04: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