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 675–697

indexedRowScatterSM86ObligationFor

Full file
675def indexedRowScatterSM86ObligationFor =
676  (lambda unrestricted geometry : (family IndexedRowScatterSM86Geometry) .
677    (eliminate
678      IndexedRowScatterSM86Geometry
679      (lambda unrestricted current : (family IndexedRowScatterSM86Geometry) .
680        (family NativeObligation))
681      geometry
682      (branch
683        IndexedRowScatterPositiveClassGradientRows
684        .
685        (constructor NativeObligation NativeClassRowScatter256))
686      (branch
687        IndexedRowScatterNegativeClassGradientRows
688        .
689        (constructor NativeObligation NativeClassRowScatter256))
690      (branch
691        IndexedRowScatterTokenEmbeddingGradientRows
692        .
693        (constructor NativeObligation NativeEmbeddingScatter1024))
694      (branch
695        IndexedRowScatterTokenEmbeddingGradientRows512x1024
696        .
697        (constructor NativeObligation NativeEmbeddingScatter512))))

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.