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 2372–2382

skProducts

Full file
2372def skProducts = (lambda unrestricted initialize : Nat . (lambda unrestricted accumulator : (pi unrestricted nt : Nat . Nat) . (lambda unrestricted a : (pi unrestricted k : Nat . Nat) .
2373  (lambda unrestricted slices : Nat . (lambda unrestricted tiles : Nat . (lambda unrestricted barrier : (family SM86Barrier) .
2374  (lambda unrestricted wait : Nat . (lambda unrestricted first : Nat . (lambda unrestricted tail : (family SM86Program) .
2375    (saFor slices (lambda unrestricted k : Nat .
2376      (saFor tiles (lambda unrestricted nt : Nat .
2377        (saHmmaFrom (accumulator nt) (a k) (skB tiles k nt)
2378          (naturalSelect (naturalIsZero k)
2379            (naturalSelect (naturalIsZero initialize) (accumulator nt) sm86RegisterZero)
2380            (accumulator nt)) barrier
2381          (naturalSelect (naturalIsZero nt) (naturalSelect (naturalIsZero k) first wait) saWaitNone)))))
2382      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.