AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.AdamWSM86

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

1,127 lines173 declarations39.3 KiBSHA-256 0c5cdf90be84
Download exact fileView at revision426597bda5753d86c4

def · lines 794–797

adamWF32CorrectedStepSizeOffsetOf

Full file
where the step-dependent scalars sit in the parameter block (a host that supplies them at run time writes there)
794def adamWF32CorrectedStepSizeOffsetOf =
795  (lambda unrestricted abi : (family AdamWF32SM86ABI) .
796    (eliminate AdamWF32SM86ABI (lambda unrestricted current : (family AdamWF32SM86ABI) . Nat) abi
797      (branch AdamWF32SM86ABIValue p g m v scale b1 c1 b2 c2 decay step epsilon one . step)))

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.