Mateusz Ozga 3c93d3c444 CK-Tile Grouped GEMM refactor and post PR fixes (#1756) ...

* Grouped gemm simple code refactor

* Offset invoker

* Invoke generic Run, and replace name of parrtitioner variable

* Tests fix type

* Removed namespaces

* Add template param to avoid implicit cast

* Remove generic function

* Constant value

* underline enum to int16_t

* Generalize partitioner function

* Remove whitespaces

* Rename function

* Using support

* Clang-format

* Clang-format

* Fn-partitioner description fn

* Typo

* Typo 2

* Better description

* Better description

* Refactor after review

* Use ctr instead of set fn

* Inovke ctr and typo

* Comments

* Remove unnecessary comment

* Review, remove modulo

2025-01-21 21:06:10 +01:00

..

add_rmsnorm2d_rdquant

support max3 in smoothquant and add+ rmsnorm + rdquant (#1654)

2024-11-27 05:01:15 +08:00

common

[Ck tile] layernorm2d fwd optimize (#1637)

2024-11-08 12:28:23 +08:00

elementwise

[CK_TILE] Fix mock token id, support g1u1/g1u0 through same inline code block (#1808)

2025-01-16 17:51:10 +08:00

epilogue

[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)

2025-01-15 10:23:48 +08:00

flatmm/block

[CK_TILE] Fix mock token id, support g1u1/g1u0 through same inline code block (#1808)

2025-01-16 17:51:10 +08:00

fmha

Update for fmha_fwd qs_ks_vs pipeline (#1810)

2025-01-13 12:43:05 +08:00

fused_moe

[CK_TILE] Fix mock token id, support g1u1/g1u0 through same inline code block (#1808)

2025-01-16 17:51:10 +08:00

gemm

CK-Tile Grouped GEMM refactor and post PR fixes (#1756)

2025-01-21 21:06:10 +01:00

image_to_column

[CK_TILE] Image to Column kernel (#1532)

2024-09-27 22:57:38 +02:00

layernorm2d

[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)

2025-01-15 10:23:48 +08:00

norm_reduce

Ck tile/layernorm: implement naive reduce, opt performance (#1784)

2025-01-03 14:28:59 +08:00

permute

Remove virtual destructors from unary ops (#1610)

2024-10-30 17:42:50 +01:00

reduce/block

[Ck_tile] smoothquant (#1617)

2024-11-01 13:51:56 +08:00

rmsnorm2d

[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)

2025-01-15 10:23:48 +08:00

smoothquant

[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)

2025-01-15 10:23:48 +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] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

common.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

elementwise.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

epilogue.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

flatmm.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

fmha.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

fused_moe.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

gemm.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

image_to_column.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

layernorm2d.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

norm_reduce.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

permute.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

reduce.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

rmsnorm2d.hpp

[CK_TILE] Add Various Fusion Functions to RMSNorm (#1802)

2025-01-15 10:23:48 +08:00

smoothquant.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

softmax.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

topk_softmax.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00

topk.hpp

[CK_TILE] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)

2025-01-07 18:49:24 +08:00