solin 24dfa0dd48 draft of mxfp4 moe

2025-11-25 08:43:24 +00:00

..

reference_batched_contraction.hpp

Update pre-commit to fixed versions, run remod for ck_tile (#2895)

2025-10-16 15:29:17 -07:00

reference_batched_dropout_randval.hpp

[CK_TILE] Support f32 in FMHA (fwd and bwd) (#2836)

2025-09-27 18:03:48 +05:00

reference_batched_dropout.hpp

CK Tile FA Training kernels (#1286)

2024-06-04 13:12:45 -05:00

reference_batched_elementwise.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

reference_batched_gemm.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

reference_batched_masking.hpp

Revert "Add attn sink (#2892)" (#3250)

2025-11-20 07:55:15 -08:00

reference_batched_rotary_position_embedding.hpp

[CK_TILE] Add PagedAttention kernels (#1387)

2024-08-28 20:50:43 +08:00

reference_batched_softmax.hpp

introducing ck_tile! (#1216)

2024-04-15 19:27:12 -05:00

reference_batched_transpose.hpp

add batched_transpose implement (#1660)

2025-01-28 16:22:02 -08:00

reference_elementwise.hpp

[CK_TILE] Support for elementwise kernel (#2246)

2025-07-24 11:21:45 +02:00

reference_fused_moe.hpp

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

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

reference_gemm.hpp

[CK_TILE] Add Flatmm MX FP8 (#3208)

2025-11-20 10:35:15 +08:00

reference_grouped_conv_bwd_data.hpp

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

2025-08-20 05:29:57 -07:00

reference_grouped_conv_bwd_weight.hpp

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

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

reference_grouped_conv_fwd.hpp

[CK_TILE] Add conv fwd + bias + clamp example (#3012)

2025-10-27 18:43:09 +01:00

reference_im2col.hpp

[CK_TILE] Image to Column kernel (#1532)

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

reference_layernorm2d_fwd.hpp

[CK_TILE] layernorm support fused-quant/fused-add (#1604)

2024-10-31 14:54:53 +08:00

reference_moe_gemm.hpp

draft of mxfp4 moe

2025-11-25 08:43:24 +00:00

reference_moe_sorting.hpp

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

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

reference_permute.hpp

[CK_TILE] fused-moe first version (#1634)

2024-11-26 11:14:56 +08:00

reference_pool.hpp

[CK_TILE] Add indexing to pooling operator (Lwpck 3892) (#3013)

2025-10-29 09:58:04 +02:00

reference_reduce.hpp

General 2D Reduction Kernel (#2535)

2025-08-06 15:36:59 +02:00

reference_rmsnorm2d_fwd.hpp

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

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

reference_rowwise_quantization2d.hpp

add fp8 as dst (#1830)

2025-01-22 17:34:27 +08:00

reference_softmax.hpp

Add -gsplit-dwarf flag to reduce debug section size and fix ckProfiler link errors (#2611)

2025-08-04 11:26:08 -07:00

reference_topk.hpp

Add -gsplit-dwarf flag to reduce debug section size and fix ckProfiler link errors (#2611)

2025-08-04 11:26:08 -07:00

reference_transpose.hpp

[CK_TILE] Support for elementwise kernel (#2246)

2025-07-24 11:21:45 +02:00