ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 02:02:46 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b7a073f76935a33eed2ad962ef412360008053d4
composable_kernel/include/ck_tile/ops
History
Max Podkorytov b7a073f769 [CK-tile] unhardcode the number of LDS banks from universal gemm policy (#3130) 
Fixes LDS bank conflicts on gfx950 for universal gemm v3 pipeline

Replaces hardcoded LDS layer calculations with dynamic computation using the new architecture helpers

Adds architecture-specific helper function get_n_lds_banks()

Changes function attributes from CK_TILE_HOST_DEVICE to CK_TILE_DEVICE in universal gemm policy


[ROCm/composable_kernel commit: 04efd282cf]
2025-10-31 11:58:11 -07:00
..
add_rmsnorm2d_rdquant
Fix for Add the API to load SGPR (#2913)
2025-09-25 10:32:42 -07:00
batched_contraction
[CK Tile] contraction multi d - kernel & example (#2901)
2025-10-13 12:30:28 +02:00
batched_transpose
Fix for Add the API to load SGPR (#2913)
2025-09-25 10:32:42 -07:00
common
[CK_TILE] Improve grouped conv kernel name generation (#3028)
2025-10-30 13:19:07 +01:00
elementwise
Lwpck 3550: Implement and test fixed precision fp8 x bf8 (#2963)
2025-10-30 13:36:10 +01:00
epilogue
[CK_TILE] Improve grouped conv kernel name generation (#3028)
2025-10-30 13:19:07 +01:00
flatmm
[CK_TILE] Add mxfp4 flatmm (#3080)
2025-10-31 11:29:05 +08:00
fmha
[CK-tile] unhardcode the number of LDS banks from universal gemm policy (#3130)
2025-10-31 11:58:11 -07:00
fused_moe
Felix/opt sorting (#2902)
2025-10-15 09:24:03 +08:00
gemm
[CK-tile] unhardcode the number of LDS banks from universal gemm policy (#3130)
2025-10-31 11:58:11 -07:00
gemm_quant
[CK_TILE] Add Bquant to Grouped Gemm (#3063)
2025-10-28 10:20:24 -04:00
grouped_convolution
[CK_TILE] Improve grouped conv kernel name generation (#3028)
2025-10-30 13:19:07 +01:00
image_to_column
Fix for Add the API to load SGPR (#2913)
2025-09-25 10:32:42 -07:00
layernorm2d
Fix for Add the API to load SGPR (#2913)
2025-09-25 10:32:42 -07:00
norm_reduce
[CK_TILE] Fix compilation errors introduced in #2320, #2219 and #2214 (#2388)
2025-06-23 12:29:15 +08:00
permute
Support Wave32 in CK_TILE - Part 1 (#2594)
2025-08-18 10:08:31 -07:00
pooling
[CK_TILE] Add indexing to pooling operator (Lwpck 3892) (#3013)
2025-10-29 09:58:04 +02:00
reduce
[CK_TILE] Add indexing to pooling operator (Lwpck 3892) (#3013)
2025-10-29 09:58:04 +02:00
rmsnorm2d
Introduce tree reduction for BlockReduce2dCrossWarpSync (#2588)
2025-10-22 14:41:35 +08:00
smoothquant
Fix for Add the API to load SGPR (#2913)
2025-09-25 10:32:42 -07:00
softmax/block
topk_softmax (#1592)
2024-10-26 23:52:49 +08:00
topk/block
topk_softmax (#1592)
2024-10-26 23:52:49 +08:00
topk_softmax
[CK_TILE] Top-K with Sigmoid kernel (#3062)
2025-10-28 10:54:06 -07:00
add_rmsnorm2d_rdquant.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
batched_contraction.hpp
Update pre-commit to fixed versions, run remod for ck_tile (#2895)
2025-10-16 15:29:17 -07:00
batched_transpose.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
common.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
elementwise.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
epilogue.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
flatmm.hpp
[CK_TILE] Add mxfp4 flatmm (#3080)
2025-10-31 11:29:05 +08:00
fmha.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
fused_moe.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
gemm_quant.hpp
Update pre-commit to fixed versions, run remod for ck_tile (#2895)
2025-10-16 15:29:17 -07:00
gemm.hpp
Pre-commit in CI (#3029)
2025-10-17 09:28:38 -07:00
grouped_convolution.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
image_to_column.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
layernorm2d.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
moe_flatmm.hpp
[CK_TILE] Update flatmm related kernels (#3022)
2025-10-22 22:36:11 +08:00
norm_reduce.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
permute.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
pooling.hpp
Update pre-commit to fixed versions, run remod for ck_tile (#2895)
2025-10-16 15:29:17 -07:00
reduce.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
rmsnorm2d.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
smoothquant.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
softmax.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
topk_softmax.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
topk.hpp
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00