AlphaGitHub
Source/Packages

Realization.Nvidia.SM86.LinearStepPlanSM86

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

348 lines59 declarations18.5 KiBSHA-256 ff3ec4d2f314
Download exact fileView at revisionbcc11a4e7647d31628

def · lines 270–281

linearStepPlanRegionsForProgram

Full file
the plan for a shape and a program for it -- the generator's, one a proposer rewrote, or another realization's (the looped one): the program region is the program's; the same offsets and launch shape. Keep this typed until whole-program encoding: arena placement must be checkable without evaluating a build-only machine-image intrinsic.
270def linearStepPlanRegionsForProgram =
271  (lambda unrestricted program : (family SM86Program) .
272    (constructor NvidiaDeviceRegions NvidiaDeviceRegionsNext
273      (constructor NvidiaDeviceRegion NvidiaDeviceProgramRegion
274        linearStepSM86Identity
275        program
276        (sm86RegisterDemand program)
277        linearStepSM86BlockWidth
278        zero
279        (constructor NvidiaDeviceRealization NvidiaDeviceRealizationRefused
280          b"linear step requires an established SM86 realization"))
281      (constructor NvidiaDeviceRegions NvidiaDeviceRegionsEnd)))

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.