AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.SoftmaxSM86

packages/realizations/cooperative/nvidia-sm86/src/Realization/Nvidia/SM86/SoftmaxSM86.alpha

2,114 lines259 declarations74.8 KiBSHA-256 5aa09b49868d
Download exact fileView at revisionf5034773ad4bb52133

def · lines 1481–1488

softmaxSM86CausalForwardProgram

Full file
1481def softmaxSM86CausalForwardProgram : (family SM86Program) =
1482  (sm86ProgramAppend
1483    softmaxSM86CausalPrefix
1484    (sm86ProgramAppend
1485      softmaxSM86MaximumReduction
1486      (sm86ProgramAppend
1487        softmaxSM86ExponentProgram
1488        (sm86ProgramAppend softmaxSM86SumReduction softmaxSM86ForwardSuffix))))

The compiler supplied declaration spans and resolved links from this source snapshot. This page does not assert that this file belongs to a checked closure.