Skip to content

feat(cake_rmsnorm_train): add fused BF16 RMSNorm training forward/backward (FP32 rstd, deterministic dw) for SM100/SM103/SM107 - #5741

Merged
yyihuang merged 1 commit into
flashinfer-ai:mainfrom
yyihuang:cake-rmsnorm-train
Oct 5, 2026
Merged

yyihuang merged 1 commit into
flashinfer-ai:mainfrom
yyihuang:cake-rmsnorm-train

cake_rmsnorm_train: re-export as an architecture-neutral module from …

c8e4218
Select commit
Loading
Failed to load commit list.
Sign in for the full log view

Annotations

2 warnings and 1 notice
label
succeeded Oct 5, 2026 in 5s