Skip to content

benchmarks.SaturnNPU.kernel_library.rms_norm

Source: benchmarks/SaturnNPU/kernel_library/rms_norm.py

benchmarks.SaturnNPU.kernel_library.rms_norm

test()

RMS norm for a split bf16 32x32 tile with preloaded inv_n and eps.