Source/Packages

Realization.Nvidia.SM86.StreamingAttentionSM86

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

3,230 lines493 declarations193.3 KiBSHA-256 cc0366b3d857

def · lines 443–450

saCountDownTile

Full file
order 1: tile = tiles - 1 - z, once the special registers are in; saCount is free until the loop count is formed from the tile
443def saCountDownTile = (lambda unrestricted order : Nat . (lambda unrestricted tiles : Nat .
444  (lambda unrestricted tail : (family SM86Program) .
445    (nat-eliminate (lambda unrestricted mode : Nat . (family SM86Program))
446      tail
447      (lambda unrestricted predecessor : Nat . (lambda unrestricted induction : (family SM86Program) .
448        (saMovImm saCount (naturalSaturatingSubtract tiles 1)
449          (saImad saTileIndex saTileIndex 4294967295 saCount tail))))
450      order))))

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.