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 580–595

exactCrossEntropyRowsProgramForUnchecked

Full file
580def exactCrossEntropyRowsProgramForUnchecked =
581  (lambda unrestricted vocabulary : Nat .
582    (sm86ProgramAppend
583      (exactCrossEntropyRowsPrefixFor vocabulary)
584      (sm86ProgramAppend
585        (exactCrossEntropyRowsMaximumTilesFor
586          (naturalDivideUnchecked vocabulary reductionSM86WarpSum256RequiredThreads))
587        (sm86ProgramAppend
588          exactCrossEntropyRowsMaximumReduction
589          (sm86ProgramAppend
590            exactCrossEntropyRowsSumPrefix
591            (sm86ProgramAppend
592              (exactCrossEntropyRowsSumTilesFor
593                (naturalDivideUnchecked vocabulary reductionSM86WarpSum256RequiredThreads))
594              (sm86ProgramAppend exactCrossEntropyRowsSumReduction
595                exactCrossEntropyRowsSuffix)))))))

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.