ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-14 18:17:44 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
4643cdd962e1bba9bb85f1d78fe1b8cbe349137f
composable_kernel/include/ck_tile/ops
History
jakpiase 4643cdd962 [CK_TILE] Switch into universal gemms for conv bwds (#2981) 
* switch into universal gemms for conv bwds

* some fixes and support universal gemm in conv fwd

* add reviewer comments

[ROCm/composable_kernel commit: 6deaaa92cc]
2025-10-14 16:09:16 +02: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
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
elementwise
[CK_TILE]enhance elementwise test (#2683)
2025-09-30 08:29:37 -07:00
epilogue
[CK Tile] CShuffle Tile Permute N all warp compatible (#2966)
2025-10-03 09:46:13 -07:00
flatmm
Fix for Add the API to load SGPR (#2913)
2025-09-25 10:32:42 -07:00
fmha
[CK_TILE] Support f32 in FMHA (fwd and bwd) (#2836)
2025-09-27 18:03:48 +05:00
fused_moe
Use __builtin_amdgcn_readfirstlane for buffer resource in fused_moe (#2893)
2025-09-30 15:12:30 -07:00
gemm
[CK_TILE] Non-K Major from old CK to CK-Tile (#2442)
2025-10-13 14:27:02 +02:00
gemm_quant
supporting prefill shapes for preshuffle block scale gemm (#2975)
2025-10-10 15:36:24 -07:00
grouped_convolution
[CK_TILE] Switch into universal gemms for conv bwds (#2981)
2025-10-14 16:09:16 +02: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] Pooling FWD (Lwpck 3683) (#2956)
2025-10-09 16:13:26 +02:00
reduce
[CK_TILE] Correct BlockWarps calculation and fix smoke-test in rmsnorm (#2540)
2025-10-13 11:52:37 -07:00
rmsnorm2d
[CK_TILE] Correct BlockWarps calculation and fix smoke-test in rmsnorm (#2540)
2025-10-13 11:52:37 -07: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
Fix for Add the API to load SGPR (#2913)
2025-09-25 10:32:42 -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
[CK Tile] contraction multi d - kernel & example (#2901)
2025-10-13 12:30:28 +02: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
Update include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02: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 include path to break the remod's cyclic dep issue (#2978)
2025-10-13 13:24:47 +02:00
gemm.hpp
[CK_TILE] Blockwise GEMM pipeline v6 - port of v5 from old CK (#2955)
2025-10-13 13:57:37 +02: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
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
pool.hpp
[CK_TILE] Pooling FWD (Lwpck 3683) (#2956)
2025-10-09 16:13:26 +02: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