3 people 86173ad593 [Kernel] Support decode context parallelism on Blackwell with CUTLASS MLA (#24385) ...

Signed-off-by: Ming Yang <minos.future@gmail.com>
Signed-off-by: youkaichao <youkaichao@gmail.com>
Co-authored-by: youkaichao <youkaichao@gmail.com>

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

..

mla

[Kernel] Support decode context parallelism on Blackwell with CUTLASS MLA (#24385)

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

attention_dtypes.h

Enable scaled FP8 (e4m3fn) KV cache on ROCm (AMD GPU) (#3290)

2024-04-03 14:15:55 -07:00

attention_generic.cuh

[CI/Build] Enforce style for C++ and CUDA code with clang-format (#4722)

2024-05-22 07:18:41 +00:00

attention_kernels.cuh

[Bugfix][ROCm] Fix for warp_size uses on host (#21205)

2025-07-24 00:37:19 -07:00

attention_utils.cuh

[AMD][CI/Build] Disambiguation of the function call for ROCm 6.2 headers compatibility (#7477)

2024-08-21 16:47:36 -07:00

dtype_bfloat16.cuh

[CI/Build] Suppress divide-by-zero and missing return statement warnings (#7001)

2024-08-05 16:00:01 -04:00

dtype_float16.cuh

[CI/Build] Enforce style for C++ and CUDA code with clang-format (#4722)

2024-05-22 07:18:41 +00:00

dtype_float32.cuh

[CI/Build] Enforce style for C++ and CUDA code with clang-format (#4722)

2024-05-22 07:18:41 +00:00

dtype_fp8.cuh

[CI/Build] Enforce style for C++ and CUDA code with clang-format (#4722)

2024-05-22 07:18:41 +00:00

merge_attn_states.cu

[BugFix] FA2 MLA Accuracy Issue (#18807)

2025-05-28 08:59:39 +00:00

paged_attention_v1.cu

[Bugfix][ROCm] Fix for warp_size uses on host (#21205)

2025-07-24 00:37:19 -07:00

paged_attention_v2.cu

[Bugfix][ROCm] Fix for warp_size uses on host (#21205)

2025-07-24 00:37:19 -07:00

vertical_slash_index.cu

Implements dual-chunk-flash-attn backend for dual chunk attention with sparse attention support (#11844)

2025-05-12 19:52:47 -07:00