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 1495–1505

softmaxSM86ProgramFor

Full file
1495def softmaxSM86ProgramFor =
1496  (lambda unrestricted variant : (family SoftmaxSM86Variant) .
1497    (eliminate
1498      SoftmaxSM86Variant
1499      (lambda unrestricted current : (family SoftmaxSM86Variant) . (family SM86Program))
1500      variant
1501      (branch SoftmaxSM86Forward . softmaxSM86ForwardProgram)
1502      (branch SoftmaxSM86CausalForward . softmaxSM86CausalForwardProgram)
1503      (branch SoftmaxSM86CausalForward1024 . softmaxSM86CausalForward1024Program)
1504      (branch SoftmaxSM86CausalBackward1024 . softmaxSM86CausalBackward1024Program)
1505      (branch SoftmaxSM86Backward . softmaxSM86BackwardProgram)))

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.