AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.AdamWHalfSM86

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

428 lines73 declarations26.3 KiBSHA-256 9692515bc485
Download exact fileView at revisionbcc11a4e7647d31628

def · lines 306–311

ahWideBody

Full file
306def ahWideBody = (lambda unrestricted wide : Nat . (lambda unrestricted stride : Nat . (lambda unrestricted quads : Nat .
307  (ahwAddresses wide stride quads
308    (ahwLoads wide stride
309      (saFor (naturalMultiply 4 wide)
310        (lambda unrestricted k : Nat . (ahwUpdate wide (naturalDivideUnchecked k 4) (naturalModuloUnchecked k 4)))
311        (ahwStores wide stride sm86ProgramEmpty)))))))

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.