AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.IndexedRowScatterSM86

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

926 lines148 declarations38.7 KiBSHA-256 c86946ffe8c6
Download exact fileView at revisionf5034773ad4bb52133

def · lines 433–443

indexedRowScatterSM86WidthImmediate

Full file
433def indexedRowScatterSM86WidthImmediate =
434  (lambda unrestricted geometry : (family IndexedRowScatterSM86Geometry) .
435    (eliminate
436      IndexedRowScatterSM86Geometry
437      (lambda unrestricted current : (family IndexedRowScatterSM86Geometry) .
438        (family SM86Unsigned32))
439      geometry
440      (branch IndexedRowScatterPositiveClassGradientRows . indexedRowScatterSM86U256)
441      (branch IndexedRowScatterNegativeClassGradientRows . indexedRowScatterSM86U256)
442      (branch IndexedRowScatterTokenEmbeddingGradientRows . indexedRowScatterSM86U1024)
443      (branch IndexedRowScatterTokenEmbeddingGradientRows512x1024 . indexedRowScatterSM86U512)))

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.