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 1459–1470

softmaxSM86CausalForward1024Program

Full file
1459def softmaxSM86CausalForward1024Program : (family SM86Program) =
1460  (sm86ProgramAppend
1461    softmaxSM86Causal1024Prefix
1462    (sm86ProgramAppend
1463      softmaxSM86LocalMaximum1024
1464      (sm86ProgramAppend
1465        softmaxSM86MaximumReduction
1466        (sm86ProgramAppend
1467          softmaxSM86Exponent1024Program
1468          (sm86ProgramAppend
1469            softmaxSM86LocalSum1024
1470            (sm86ProgramAppend softmaxSM86SumReduction softmaxSM86Suffix1024))))))

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.