Hanjie Qiu dcb28a332b

[Kernel] Flashinfer MLA (trtllm-gen) decode kernel integration (#21078) ...

Signed-off-by: hjjq <hanjieq@nvidia.com>
Signed-off-by: Michael Goin <mgoin64@gmail.com>
Signed-off-by: mgoin <mgoin64@gmail.com>
Co-authored-by: Michael Goin <mgoin64@gmail.com>

2025-09-10 15:31:10 -07:00

..

conftest.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_aiter_flash_attn.py

[CI Perf] Prune tests in tests/kernels/attention/ (#22936)

2025-08-14 21:34:53 -06:00

test_attention_selector.py

[Attention] FlashAttn MLA (#14258)

2025-09-04 02:47:59 -07:00

test_attention.py

[CI Perf] Prune tests in tests/kernels/attention/ (#22936)

2025-08-14 21:34:53 -06:00

test_cache.py

[Kernel] cuda kernels for upcoming decode context parallel feature (#23791)

2025-08-28 15:29:11 +08:00

test_cascade_flash_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_cutlass_mla_decode.py

[Kernel] Flashinfer MLA (trtllm-gen) decode kernel integration (#21078)

2025-09-10 15:31:10 -07:00

test_encoder_decoder_attn.py

[CI] change spell checker from codespell to typos (#18711)

2025-06-11 19:57:10 -07:00

test_flash_attn.py

[CI Perf] Prune tests in tests/kernels/attention/ (#22936)

2025-08-14 21:34:53 -06:00

test_flashinfer_mla_decode.py

[Kernel] Flashinfer MLA (trtllm-gen) decode kernel integration (#21078)

2025-09-10 15:31:10 -07:00

test_flashinfer_trtllm_attention.py

[Flashinfer] Support Flashinfer TRTLLM FP8-qkv BF16/FP16-out Attention Kernel (#23647)

2025-09-08 20:53:07 -07:00

test_flashinfer.py

[Core] Always use tensor cores for Flashinfer Decode Wrapper (#23214)

2025-08-21 16:02:11 -04:00

test_flashmla.py

[Kernel] Add FP8 support with FlashMLA backend (#22668)

2025-08-22 02:26:32 +00:00

test_lightning_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_merge_attn_states.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -07:00

test_mha_attn.py

Feature/vit attention unification# 23880 (#23978)

2025-09-10 06:10:14 -07:00

test_mla_decode_cpu.py

[Refactor] Remove duplicate ceil_div (#20023)

2025-06-25 05:19:09 +00:00

test_prefix_prefill.py

[CI Perf] Prune tests in tests/kernels/attention/ (#22936)

2025-08-14 21:34:53 -06:00

test_rocm_attention_selector.py

[V0 Deprecation] Deprecate BlockSparse Attention & Phi3-Small (#21217)

2025-07-19 13:53:17 -07:00

test_triton_decode_attention.py

[Refactor] Remove duplicate ceil_div (#20023)

2025-06-25 05:19:09 +00:00

test_triton_unified_attention.py

[CI Perf] Prune tests in tests/kernels/attention/ (#22936)

2025-08-14 21:34:53 -06:00