3 people 100b630a60 [V1][Kernel] Add triton implementation for reshape_and_cache_flash (#24503) ...

Signed-off-by: Burkhard Ringlein <ngl@zurich.ibm.com>
Co-authored-by: Chih-Chieh Yang <chih.chieh.yang@ibm.com>
Co-authored-by: Wentao Ye <44945378+yewentao256@users.noreply.github.com>

2025-09-23 12:52:40 -04: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

[CI/Build] Fix disabled v1 attention backend selection test (#25471)

2025-09-23 13:05:46 +00:00

test_attention.py

Remove V0 attention backends (#25351)

2025-09-21 16:03:28 -07:00

test_cache.py

[V1][Kernel] Add triton implementation for reshape_and_cache_flash (#24503)

2025-09-23 12:52:40 -04: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_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

[Multi Modal] Add FA3 in VIT (#24347)

2025-09-12 21:27:24 +08:00

test_mla_decode_cpu.py

[Refactor] Remove duplicate ceil_div (#20023)

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

test_prefix_prefill.py

Remove V0 attention backends (#25351)

2025-09-21 16:03:28 -07:00

test_rocm_attention_selector.py

Remove V0 attention backends (#25351)

2025-09-21 16:03:28 -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

Optimize triton unified attention performance for sliding window attention (#24390)

2025-09-19 13:07:26 -06:00