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 249–259

ahwAddresses

Full file
bounds (quad q's guard: T + q S at or past `quads`), then the pointers
249def ahwAddresses = (lambda unrestricted wide : Nat . (lambda unrestricted stride : Nat . (lambda unrestricted quads : Nat .
250  (lambda unrestricted tail : (family SM86Program) .
251  (saGreater saP0 2 (naturalPredecessor quads)
252  (saFor (naturalPredecessor wide) (lambda unrestricted j : Nat . (lambda unrestricted rest : (family SM86Program) .
253    (saAddImm 27 2 (naturalMultiply (succ j) stride)
254    (saGreater (ahwPredicate (succ j)) 27 (naturalPredecessor quads) rest))))
255  (saWide 6 2 3 (saArgument adamWHalfSM86ParameterArgument)
256  (saWide 8 2 3 (saArgument adamWHalfSM86GradientArgument)
257  (saWide 10 2 3 (saArgument adamWHalfSM86FirstMomentArgument)
258  (saWide 12 2 3 (saArgument adamWHalfSM86SecondMomentArgument)
259  (saWide 30 2 4 ahHalfPointer 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.