qizixi a2a5f79e09

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

Signed-off-by: zixi-qi <qizixi@meta.com>

2025-09-19 13:07:26 -06: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 Bugfix] Fix failing test_invalid_env (#25078)

2025-09-17 08:28:58 -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_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

[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

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

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