AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.ExactCrossEntropySM86

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

1,471 lines228 declarations53.2 KiBSHA-256 411308c1414a
Download exact fileView at revisionf5034773ad4bb52133

def · lines 982–988

exactCrossEntropyFinalizeProgramForUnchecked

Full file
982def exactCrossEntropyFinalizeProgramForUnchecked =
983  (lambda unrestricted rows : Nat .
984    (sm86ProgramAppend
985      (exactCrossEntropyFinalizePrefixFor rows
986        (naturalDivideUnchecked rows reductionSM86WarpSum1024RequiredThreads))
987      (sm86ProgramAppend exactCrossEntropyFinalizeReduction
988        exactCrossEntropyFinalizeSuffix)))

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.