ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-01 04:07:56 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
flatmm-moe-remap-xcd
composable_kernel/include/ck_tile/ops
History
Juuso Korhonen da0a56da24 add commenting
2025-09-10 12:21:12 +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
support swiglu activaion and use rcpf to accelerate silu
2025-08-26 12:32:29 +00:00
epilogue
Refactor scale and bias function, encapsulate scale/bias pointer to a tensor view and load data by tilewise operation
2025-09-09 03:52:03 +00:00
flatmm
Refactor scale and bias function, encapsulate scale/bias pointer to a tensor view and load data by tilewise operation
2025-09-09 03:52:03 +00:00
fmha
Add separate mask checking for scope [aligned_physical_seqlen_k_start, physical_seqlen_k_end) (#2487)
2025-07-11 18:14:47 +08:00
fused_moe
[CK_TILE] moe sorting optimize local_token (#2469)
2025-07-15 09:42:18 +08:00
gemm
fixed remap
2025-09-10 11:11:29 +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
moe_flatmm
add commenting
2025-09-10 12:21:12 +00: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
debug mixed_prec flatmm
2025-08-07 09:22:04 +00:00
fmha.hpp
ck tile pagedkv prefill (#2405)
2025-07-07 16:16:54 +08: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
moe_flatmm.hpp
add moe_flatmm
2025-08-06 08:33:33 +00: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