Source/Packages

Realization.Nvidia.SM86.AdamWHalfSM86

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

428 lines73 declarations26.3 KiBSHA-256 9692515bc485

def · lines 84–88

ahScalarsFrom

Full file
the scalars from the constant bank into R18 + j, from j = `first` on
84def ahScalarsFrom = (lambda unrestricted first : Nat . (lambda unrestricted tail : (family SM86Program) .
85  (saFor (naturalSaturatingSubtract 8 first)
86    (lambda unrestricted j : Nat .
87      (saMovConst (naturalAdd 18 (naturalAdd first j)) (ahScalar (naturalAdd first j))))
88    tail)))

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.