Expand description
Softmax kernel: numerically stable exponential normalization.
Matches softmax-kernel-v1.yaml.
Four phases: find_max -> exp_subtract -> sum_exp -> normalize.
Functions§
- softmax_
avx2 ⚠ - AVX2 SIMD implementation of numerically stable softmax.
- softmax_
ptx - PTX assembly for softmax kernel.
- softmax_
scalar - Scalar reference implementation of numerically stable softmax.