ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-11 17:00:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
94b6430489a7be3611234322a9e1b88ebcf0564f
composable_kernel/include/ck_tile/ops
History
aska-0096 94b6430489 temp save
2025-07-17 10:06:09 +00:00
..
add_rmsnorm2d_rdquant
[A8W8 GEMM] Optimized weight-preshuffled implementation & add quantization datatype for CK TILE rms_norm (#1862)
2025-02-20 14:00:27 -08:00
batched_transpose
Vectorized Transpose for Batched Transpose CK Tile Operator (#2131)
2025-05-12 00:41:45 -07:00
common
chore: unset executable permission (#2303)
2025-06-10 09:13:59 -07:00
elementwise
[CK_TILE] Multiple-D GEMM example (#2219)
2025-06-13 19:39:11 +02:00
epilogue
[CK TILE] Grouped Convolution Forward Kernel (#2188)
2025-06-20 15:44:36 -07:00
flatmm
Merge flatmm Operator with universal gemm (#2434)
2025-07-11 08:27:55 -07:00
fmha
temp save
2025-07-17 10:06:09 +00:00
fused_moe
[CK_TILE] moe sorting optimize local_token (#2469)
2025-07-15 09:42:18 +08:00
gemm
temp save
2025-07-17 10:06:09 +00:00
grouped_convolution
[CK TILE] Grouped Convolution Forward Kernel (#2188)
2025-06-20 15:44:36 -07:00
image_to_column
Do not use warpSize as compile time constant as it is removed (#2320)
2025-06-17 11:54:30 -07:00
layernorm2d
[CK_TILE] Improve RMS/Layer Normalization 2 Pass Pipeline Performance (#1861)
2025-03-25 20:09:45 +08:00
norm_reduce
[CK_TILE] Fix compilation errors introduced in #2320, #2219 and #2214 (#2388)
2025-06-23 12:29:15 +08:00
permute
Remove virtual destructors from unary ops (#1610)
2024-10-30 17:42:50 +01:00
reduce/block
[CK_TILE] Enhance RMSNorm Accuracy: New Pipeline Pass for Selectable Implementation (#2409)
2025-07-16 14:05:26 +08:00
rmsnorm2d
[CK_TILE] Enhance RMSNorm Accuracy: New Pipeline Pass for Selectable Implementation (#2409)
2025-07-16 14:05:26 +08:00
smoothquant
add fp8 as dst (#1830)
2025-01-22 17:34:27 +08: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
topk_softmax (#1592)
2024-10-26 23:52:49 +08:00
add_rmsnorm2d_rdquant.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
batched_transpose.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
common.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
elementwise.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
epilogue.hpp
Solve the Static Encoding Pattern compile error when the tile size is too small (#2079)
2025-04-13 20:09:30 -07:00
flatmm.hpp
[flatmm] implement basic fp16 flatmm (#2089)
2025-04-16 16:51:17 +08:00
fmha.hpp
Merge branch 'test_copy_fix' of https://github.com/ROCm/composable_kernel into fa_decode_pipeline
2025-07-17 07:24:32 +00:00
fused_moe.hpp
[CK_TILE] moe sorting ex kernel to support expert > 128 (#1840)
2025-02-11 17:49:17 +08:00
gemm.hpp
Merge flatmm Operator with universal gemm (#2434)
2025-07-11 08:27:55 -07:00
grouped_convolution.hpp
[CK TILE] Grouped Convolution Forward Kernel (#2188)
2025-06-20 15:44:36 -07:00
image_to_column.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
layernorm2d.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
norm_reduce.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
permute.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
reduce.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
rmsnorm2d.hpp
[CK_TILE] Enhance RMSNorm Accuracy: New Pipeline Pass for Selectable Implementation (#2409)
2025-07-16 14:05:26 +08:00
smoothquant.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
softmax.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
topk_softmax.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00
topk.hpp
[CK_TILE] Add GetName for GEMM kernels (#1791)
2025-02-11 09:49:48 +01:00