Khushbu Agarwal 3c39d279ab supporting prefill shapes for preshuffle block scale gemm (#2975) ...

* debugging

* debugging for prefill shapes

* comment unused code

* fix for prefill shapes

* clearing up the code

* add int4 to universal gemm example

* clang formatted

* adding test for prefill shapes in block scale gemm

* lil improv on the block pipeline

* Address Review Comment

---------

Co-authored-by: ThomasNing <thomas.ning@amd.com>

2025-10-10 15:36:24 -07:00

..

add_rmsnorm2d_rdquant

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07:00

batched_transpose

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07:00

common

Add gemm weight preshuffle pk_int_t support (#2858)

2025-09-18 21:26:10 -07: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] functional support for transposed inputs in compute-bound double-lds-buffer pipeline with async loads from global memory to LDS (#2984)

2025-10-10 12:57:50 -07:00

gemm_quant

supporting prefill shapes for preshuffle block scale gemm (#2975)

2025-10-10 15:36:24 -07:00

grouped_convolution

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07: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

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -07:00

rmsnorm2d

Fix for Add the API to load SGPR (#2913)

2025-09-25 10:32:42 -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

[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] Stream-K tile partitioner (#2708)

2025-09-03 13:38:17 -07: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

[CK_TILE] Add FAv3 fwd pipeline (#2731)

2025-09-01 09:16:45 +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_quant.hpp

Add Memory pipeline for AQuant Block Scale GEMM (#2987)

2025-10-08 17:22:30 -07:00

gemm.hpp

Add a new gemm pipeline based on ComputeV4 which utilizes async copy API (#2949)

2025-10-01 15:38:07 -07:00

grouped_convolution.hpp

[CK Tile] Grouped convolution backward data (#2652)

2025-08-20 05:29:57 -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

pool.hpp

[CK_TILE] Pooling FWD (Lwpck 3683) (#2956)

2025-10-09 16:13:26 +02: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