Pavani Majety 1d353b6352 [Core] Always use tensor cores for Flashinfer Decode Wrapper (#23214) ...

Signed-off-by: Pavani Majety <pmajety@nvidia.com>

2025-08-21 16:02:11 -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

[V0 Deprecation] Remove V0 FlashInfer attention backend (#22776)

2025-08-18 19:54:16 -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

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

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

test_cascade_flash_attn.py

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -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_trtllm_attention.py

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

2025-08-21 16:02:11 -04: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

[Misc] parametrize 'dtype' in test_flash_mla (#22641)

2025-08-12 16:31:48 -04: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

[Misc] Add SPDX-FileCopyrightText (#19100)

2025-06-03 11:20:17 -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