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 402–407

adamWHalfSM86ScaledSide

Full file
the scaled image on side warps: `threads` threads a CTA past its first `first`, over a grid `gridX` wide, `stride` the side threads' count
402def adamWHalfSM86ScaledSide = (lambda unrestricted finish : (pi unrestricted piece : (family SM86Program) . (family SM86Program)) .
403  (lambda unrestricted wide : Nat .
404  (lambda unrestricted first : Nat . (lambda unrestricted threads : Nat . (lambda unrestricted gridX : Nat .
405  (lambda unrestricted quads : Nat . (lambda unrestricted stride : Nat .
406    (ahAssembleBody (ahSidePrologueWith 1 first threads gridX) (ahWideBody wide stride quads)
407      finish 1 quads (naturalMultiply wide stride)))))))))

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.