Yashvardhan Agarwal b13aa4217e CK_TILE: Implement two-stage split-K GEMM with workspace reduction (LWPCK-2966) (#2632) ...

* CK_TILE: Implement two-stage split-K GEMM with reduction

- Added split-K GEMM with reduction example

* comment resolutions

[ROCm/composable_kernel commit: 7f14772406]

2025-08-14 10:18:52 +02: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

[CK-tile] add more tests for batched transpose testing the rectangular block tile sizes (#2634)

2025-08-07 16:51:53 -07:00

common

chore: unset executable permission (#2303)

2025-06-10 09:13:59 -07:00

elementwise

[CK_TILE]fix elementwise example in gfx11/12 (#2676)

2025-08-13 15:21:46 -07:00

epilogue

[CK_TILE] FMHA BWD Decode Pipeline (#2643)

2025-08-12 17:02:52 +08:00

flatmm

[CK_TILE]fix ck_tile's moe_sorting example in gfx11 (#2667)

2025-08-12 12:33:56 -07:00

fmha

Cleanups (#2631)

2025-08-13 10:12:08 +02:00

fused_moe

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -07:00

gemm

CK_TILE: Implement two-stage split-K GEMM with workspace reduction (LWPCK-2966) (#2632)

2025-08-14 10:18:52 +02:00

gemm_group_quant

Preshuffle AQ matrix in block scale gemm (#2624)

2025-08-12 21:32:51 -07:00

grouped_convolution

upgrade from clang-format-12 to clang-format-18 (#2568)

2025-07-28 11:34:07 -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

Re-enable optimization for gfx950 fmha fwd (#2671)

2025-08-13 14:57:43 +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] Merge transpose examples (#2450)

2025-07-26 21:51:54 -07:00

common.hpp

[CK_TILE] Add GetName for GEMM kernels (#1791)

2025-02-11 09:49:48 +01:00

elementwise.hpp

[CK_TILE] Support for elementwise kernel (#2246)

2025-07-24 11:21:45 +02: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

Minor Improvements in CK TILE memory copy EXAMPLE (#2678)

2025-08-13 15:24:16 -07:00

fused_moe.hpp

[CK_TILE] moe sorting ex kernel to support expert > 128 (#1840)

2025-02-11 17:49:17 +08:00

gemm_group_quant.hpp

[CK_TILE] Grouped Convolution Backward Weight Kernel (#2357)

2025-07-24 10:41:35 +02:00

gemm.hpp

Minor Improvements in CK TILE memory copy EXAMPLE (#2678)

2025-08-13 15:24:16 -07:00

grouped_convolution.hpp

[CK_TILE] Grouped Convolution Backward Weight Kernel (#2357)

2025-07-24 10:41:35 +02: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

GEMM Multi D for CK Tile Engine (#2660)

2025-08-12 16:05:05 -07: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