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
5b559e7409f2c90854013e01fdbbc880c69e0e55
composable_kernel/include/ck_tile/ops
History
Ville Pietilä f1c40b8fba [CK_TILE] Merge multiple fwd convolution groups into a single GEMM batch. (#3136) 
* Merge fwd conv groups in CK Tile.

* Fix building CK fwd convs.

* Add number of merged groups to conv fwd kernel name.

* Get number of merged groups from conv config.

* Rename GemmConfig to ConvConfig.

* Clean-up TODOs.

* Check that number of conv groups must be divisible by the number of merged groups.

* Improve error handling in the conv fwd example.

* Fix clang-format.

* Fix group offsets.

* Fix merge problem.

* Address feedback from code review.

* Fix clang-formatting.

[ROCm/composable_kernel commit: 66832861ad]
2025-12-02 15:23:32 +02: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
[CK_TILE] Move DataTypeTraits into a Common File (#3146)
2025-11-27 09:09:54 -08:00
elementwise
Fix and improve the gemm quant pipeline infrastructure (#3245)
2025-11-26 18:04:27 -08:00
epilogue
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
flatmm
[CK_Tile] Flatmm MX Cleanup & Explicite Offset Calculation (#3286)
2025-12-02 14:21:12 +08:00
fmha
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
fused_moe
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
gemm
[CK_TILE] Fix for comp pipeline v4 (#3307)
2025-12-02 11:38:06 +01:00
gemm_quant
Make CK TILE GEMM Aquant support block tile 128x128x128 (#3325)
2025-12-01 15:04:37 -08:00
grouped_convolution
[CK_TILE] Merge multiple fwd convolution groups into a single GEMM batch. (#3136)
2025-12-02 15:23:32 +02: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
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
permute
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
pooling
[CK Tile] enable building examples by default (#3259)
2025-11-26 16:24:44 -08:00
reduce
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
rmsnorm2d
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
smoothquant
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
softmax/block
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
topk/block
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07:00
topk_softmax
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -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
Support fp8 dynamic quantization for fmha (#3206)
2025-11-24 16:28:25 +08: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
formatting (#3182)
2025-11-11 07:42:26 -08:00
gemm.hpp
Replace CK_TILE_PIPELINE macros with a common enum
2025-11-03 09:35:05 -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
chore(copyright): update copyright header for include directory (#3293)
2025-11-26 11:00:05 -07: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