Damien Lejeune b83c07748c WIP: arbitrary batch dim

2026-01-28 06:00:10 -05:00

..

reference_batched_contraction.hpp

[CK Tile] batched contraction kernel generalizing (#3126)

2025-12-02 13:30:27 +01:00

reference_batched_dropout_randval.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_batched_dropout.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_batched_elementwise.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_batched_gemm.hpp

Fp8 block scale quantization for fmha fwd (#3330)

2026-01-21 20:58:26 -08:00

reference_batched_masking.hpp

Add attention sink support for FMHA FWD (#3368)

2025-12-15 12:21:59 +08:00

reference_batched_rotary_position_embedding.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_batched_softmax.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_batched_transpose.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_elementwise.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_fused_moe.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_gemm.hpp

Enable padding blockscale for abquant (#3453)

2025-12-24 09:12:40 -08:00

reference_grouped_conv_bwd_data.hpp

Update unsigned long literals and format specifiers to work correctly in Windows (#3483)

2026-01-02 22:16:41 -07:00

reference_grouped_conv_bwd_weight.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_grouped_conv_fwd.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_im2col.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_layernorm2d_fwd.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_mhc.hpp

WIP: arbitrary batch dim

2026-01-28 06:00:10 -05:00

reference_moe_gemm.hpp

[CK_TILE] support split-k a16w4 gemm1 (#3389)

2025-12-29 23:05:35 +08:00

reference_moe_sorting.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_permute.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_pool.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_reduce.hpp

Dlejeune/ck tile 2d multiple reductions (#3147)

2026-01-09 11:16:37 +01:00

reference_rmsnorm2d_fwd.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_rowwise_quantization2d.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_softmax.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_topk.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00

reference_transpose.hpp

chore(copyright): update copyright header for include directory (#3293)

2025-11-26 11:00:05 -07:00