benchmarks.SaturnNPU.kernel_library.softmax
Source: benchmarks/SaturnNPU/kernel_library/softmax.py
benchmarks.SaturnNPU.kernel_library.softmax
test()
Stable row softmax for a split bf16 32x32 tile.