607def sqTileBody = (lambda unrestricted seq : Nat . (lambda unrestricted tail : (family SM86Program) .
608 (saImad saMask saCount 4096 saMaskBase
609 -- S = Q K^T
610 (saWide saKeyPointer saKeyOffset saOne (saArgument 1)
611 (sqLoadRows saKeyPointer saWait3
612 (saFor 32 (lambda unrestricted i : Nat . (saMovImm (naturalAdd 36 i) 0))
613 (saFor 32 (lambda unrestricted i : Nat . (saMovImm (naturalAdd 212 i) 0))
614 (sqProducts saS (saQ 0) (naturalAdd saWait0 saWait1)
615 -- dP = dO V^T (V's rows at K's offset)
616 (saWide saKeyPointer saKeyOffset saOne (saArgument 2)
617 (sqLoadRows saKeyPointer saWait2
618 (sqProducts sqDP (sqDOut 0) saWait0
619 -- K^T's fragments for dQ, once dP's products are done
620 (saWide saValuePointer sqKTOffset saOne (saArgument 3)
621 (sqLoadKeysT seq
622 (saMaskTile
623 (sqScoreGradient
624 (saPackP
625 (saValues
626 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.