Files
composable_kernel/include/ck_tile/ops
Jeff Huang 4e06eaa417 [CK_TILE] fmha: Add query padding support to backward pass
Introduces support for query sequence padding (q_padding) in the FMHA backward pass kernels.
- Passing `seqlen_q_ptr` to the backward kernels to distinguish logical from physical sequence lengths.
- Updating `OGradDotO`, `ConvertQGrad`, and `DQDKDV` kernels to respect logical lengths and handle zero-length sequences.
- Aligning LSE indexing in the forward kernel with the padded layout for consistency.
- Adding a new GTest suite (`test_fmha_bwd_kernel_padding.cpp`) with comprehensive tests for various padding scenarios, including zero-length
  sequences and deterministic mode.
2025-10-25 16:10:35 +08:00
..
2025-10-15 09:24:03 +08:00
2024-10-26 23:52:49 +08:00
2024-10-26 23:52:49 +08:00
2025-10-17 09:28:38 -07:00