Emily Martins f87f768d16 Stream-K Tile Partitioner Base Class with Tests ...

To better align with the original Stream-K paper, this change implements
a new Stream-K tile partitioner base class. This class will handle the
Stream-K setup that is common to both a persistent and non-persistent DP
section. A later change will implement derived classes to handle the
differences between persistent and non-persistent DP.

This change also includes unit tests for the base tile partitioner.

2025-10-16 08:47:06 -06:00

..

add_rmsnorm2d_rdquant

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

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

batched_contraction

[CK Tile] contraction multi d - kernel & example (#2901)

2025-10-13 12:30:28 +02:00

batched_transpose

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

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

common

Update include path to break the remod's cyclic dep issue (#2978)

2025-10-13 13:24:47 +02: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

Felix/opt sorting (#2902)

2025-10-15 09:24:03 +08:00

gemm

Stream-K Tile Partitioner Base Class with Tests

2025-10-16 08:47:06 -06:00

gemm_quant

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

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

grouped_convolution

re-enable clang-format by default (#3030)

2025-10-15 07:43:11 -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

[CK_TILE] Correct BlockWarps calculation and fix smoke-test in rmsnorm (#2540)

2025-10-13 11:52:37 -07:00

rmsnorm2d

[CK_TILE] Correct BlockWarps calculation and fix smoke-test in rmsnorm (#2540)

2025-10-13 11:52:37 -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

Update include path to break the remod's cyclic dep issue (#2978)

2025-10-13 13:24:47 +02:00

batched_contraction.hpp

[CK Tile] contraction multi d - kernel & example (#2901)

2025-10-13 12:30:28 +02: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

Update include path to break the remod's cyclic dep issue (#2978)

2025-10-13 13:24:47 +02:00

fmha.hpp

Update include path to break the remod's cyclic dep issue (#2978)

2025-10-13 13:24:47 +02: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

Update include path to break the remod's cyclic dep issue (#2978)

2025-10-13 13:24:47 +02:00

gemm.hpp

Stream-K Tile Partitioner Base Class with Tests

2025-10-16 08:47:06 -06: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

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

pool.hpp

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

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