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 596–604

exactCrossEntropyRowsShapeAdmitted

Full file
596def exactCrossEntropyRowsShapeAdmitted =
597  (lambda unrestricted rows : Nat . (lambda unrestricted vocabulary : Nat .
598    (naturalAnd (naturalNonzero rows)
599      (naturalAnd
600        (naturalIsZero
601          (naturalModuloUnchecked vocabulary reductionSM86WarpSum256RequiredThreads))
602        (naturalLess
603          (naturalMultiply rows (naturalMultiply vocabulary 4))
604          (naturalPowerOfTwo 32))))))

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.