ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-06-30 03:37:38 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
724280378db85df35034883ec9472a1a1ed07646
composable_kernel/include/ck_tile/ops
History
aska-0096 724280378d temp save
2025-02-14 08:24:17 +00:00
..
add_rmsnorm2d_rdquant
support max3 in smoothquant and add+ rmsnorm + rdquant (#1654)
2024-11-27 05:01:15 +08:00
batched_transpose
add batched_transpose implement (#1660)
2025-01-28 16:22:02 -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
Support for dtypes (fp8, bf8, bf16 and fp16) for the ck_tile/03_gemm example. (#1845)
2025-02-07 15:07:06 -07:00
flatmm/block
Merge from internal (#1857)
2025-02-07 15:05:05 -07:00
fmha
temp save
2025-02-14 08:24:17 +00: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 - small fix to hotloop scheduler & KPack value. (#1867)
2025-02-07 15:07:07 -07: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
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] fmha fwd splitkv optimization for decode (seqlen_q=1) (#1789)
2025-01-07 18:49:24 +08:00
batched_transpose.hpp
Support for dtypes (fp8, bf8, bf16 and fp16) for the ck_tile/03_gemm example. (#1845)
2025-02-07 15:07:06 -07: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