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 422–431

indexedRowScatterSM86RowsNatural

Full file
422def indexedRowScatterSM86RowsNatural =
423  (lambda unrestricted geometry : (family IndexedRowScatterSM86Geometry) .
424    (eliminate
425      IndexedRowScatterSM86Geometry
426      (lambda unrestricted current : (family IndexedRowScatterSM86Geometry) . Nat)
427      geometry
428      (branch IndexedRowScatterPositiveClassGradientRows . indexedRowScatterSM86N6144)
429      (branch IndexedRowScatterNegativeClassGradientRows . indexedRowScatterSM86N4096)
430      (branch IndexedRowScatterTokenEmbeddingGradientRows . indexedRowScatterSM86N6144)
431      (branch IndexedRowScatterTokenEmbeddingGradientRows512x1024 . indexedRowScatterSM86N1024)))

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.