Batch norm updates and fused kernel - #1689

Draft

elliottslaughter wants to merge 6 commits into

flexflow:masterfrom

elliottslaughter:batch-norm

Commits on Sep 30, 2026

Update batch_norm kernels.

elliottslaughter
and
claude
committed
a1bd7af
Implement SiLU in the element_unary GPU kernels.

elliottslaughter
and
claude
committed
b06f3ee
Add BatchNormMode.
elliottslaughter
committed
8bdbab4
Fix silu kernel backward pass.
elliottslaughter
committed
e04b179
Compute a batch norm and the activation after it in one kernel.

elliottslaughter
and
claude
committed
0715d15
Give batch norm a grid the device can fill.

elliottslaughter
and
claude
committed
4ee795c