ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-29 11:16:59 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
mohsen-backup
composable_kernel/include/ck_tile/ops
History
msaffari-amd ee3ada6e4a [AITERKER-112] PER_TOKEN_HEAD: support page_size < kN0 via cross-page dequant 
- Pipeline: remove kPageBlockSize >= kN0 static_assert; QK dequant now
  precomputes tile_k_pages[] and indexes per-column. page_size >= kN0 stays
  on the original single-page fast path (kPagesPerTile==1).
- Codegen: add page_size=64 to SUPPORTED_PAGE_SIZE; drop per_token_head from
  the page_size < tile.F_bn0 filter (kv_blockscale still filtered).
2026-05-20 14:21:12 +00:00
..
add_rmsnorm2d_rdquant
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
batched_contraction
[CK Tile] batched contraction kernel generalizing (#3126)
2025-12-02 13:30:27 +01:00
batched_transpose
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
common
[rocm-libraries] ROCm/rocm-libraries#4769 (commit 72ae66e)
2026-04-14 17:51:20 +00:00
elementwise
[rocm-libraries] ROCm/rocm-libraries#6399 (commit 13bf528)
2026-04-14 07:45:14 +00:00
epilogue
[rocm-libraries] ROCm/rocm-libraries#5863 (commit 31d9247)
2026-04-14 20:23:26 +00:00
flatmm
[rocm-libraries] ROCm/rocm-libraries#6302 (commit 8d419e8)
2026-04-10 15:18:02 +00:00
fmha
[AITERKER-112] PER_TOKEN_HEAD: support page_size < kN0 via cross-page dequant
2026-05-20 14:21:12 +00:00
fused_moe
[rocm-libraries] ROCm/rocm-libraries#6242 (commit f46ac14)
2026-04-23 22:45:32 +00:00
gemm
[rocm-libraries] ROCm/rocm-libraries#6574 (commit b3db057)
2026-04-30 18:33:36 +00:00
gemm_mx
[rocm-libraries] ROCm/rocm-libraries#5849 (commit d9b89b2)
2026-03-27 20:37:23 +00:00
gemm_quant
[rocm-libraries] ROCm/rocm-libraries#5939 (commit 6fb1791)
2026-04-07 14:38:07 +00:00
grouped_convolution
[rocm-libraries] ROCm/rocm-libraries#5842 (commit 04c5690)
2026-03-31 08:03:41 +00:00
image_to_column
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
layernorm2d
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
norm_reduce
[rocm-libraries] ROCm/rocm-libraries#6302 (commit 8d419e8)
2026-04-10 15:18:02 +00:00
permute
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
pooling
Shuffle fix for gfx950 (#3491)
2026-01-13 09:21:29 -08:00
reduce
[rocm-libraries] ROCm/rocm-libraries#6302 (commit 8d419e8)
2026-04-10 15:18:02 +00:00
rmsnorm2d
Fix redundant cast in model sensitive rmsnorm (#3681)
2026-01-30 10:52:19 +08:00
sageattention
[rocm-libraries] ROCm/rocm-libraries#6574 (commit b3db057)
2026-04-30 18:33:36 +00:00
smoothquant
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
softmax/block
[rocm-libraries] ROCm/rocm-libraries#4274 (commit 7c380df)
2026-02-11 05:52:42 +00:00
sparse_attn
[CK_TILE][FMHA] Add sparse attention VSA (#3341)
2026-01-31 00:59:47 +08:00
topk/block
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
topk_softmax
Shuffle fix for gfx950 (#3491)
2026-01-13 09:21:29 -08:00
add_rmsnorm2d_rdquant.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
batched_contraction.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
batched_transpose.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
common.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
elementwise.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
epilogue.hpp
[rocm-libraries] ROCm/rocm-libraries#5863 (commit 31d9247)
2026-04-14 20:23:26 +00:00
flatmm.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
fmha.hpp
[rocm-libraries] ROCm/rocm-libraries#6209 (commit 89c9f3e)
2026-04-24 16:31:59 +00:00
fused_moe.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
gemm_mx.hpp
[rocm-libraries] ROCm/rocm-libraries#5241 (commit 43daeac)
2026-03-12 08:27:49 +00:00
gemm_quant.hpp
[rocm-libraries] ROCm/rocm-libraries#4964 (commit 3271d9a)
2026-03-16 08:31:56 +00:00
gemm.hpp
[rocm-libraries] ROCm/rocm-libraries#4964 (commit 3271d9a)
2026-03-16 08:31:56 +00:00
grouped_convolution.hpp
[rocm-libraries] ROCm/rocm-libraries#5241 (commit 43daeac)
2026-03-12 08:27:49 +00:00
image_to_column.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
layernorm2d.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
moe_flatmm.hpp
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
norm_reduce.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
permute.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
pooling.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
reduce.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
rmsnorm2d.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
sageattn.hpp
[rocm-libraries] ROCm/rocm-libraries#6574 (commit b3db057)
2026-04-30 18:33:36 +00:00
smoothquant.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
softmax.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
sparse_attn.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
topk_softmax.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00
topk.hpp
[rocm-libraries] ROCm/rocm-libraries#4294 (commit 6601702)
2026-03-02 12:21:44 +00:00