1315def sqTileBody = (lambda unrestricted seq : Nat . (lambda unrestricted tail : (family SM86Program) .
1316 (saImad saMask saCount 4096 saMaskBase
1317 -- S = Q K^T
1318 (saWide saKeyPointer saKeyOffset saOne (saArgument 1)
1319 (sqLoadRows saKeyPointer saWait3
1320 (saFor 32 (lambda unrestricted i : Nat . (saMovImm (naturalAdd 36 i) 0))
1321 (saFor 32 (lambda unrestricted i : Nat . (saMovImm (naturalAdd 212 i) 0))
1322 (sqProducts saS (saQ 0) (naturalAdd saWait0 saWait1)
1323 -- dP = dO V^T (V's rows at K's offset)
1324 (saWide saKeyPointer saKeyOffset saOne (saArgument 2)
1325 (sqLoadRows saKeyPointer saWait2
1326 (sqProducts sqDP (sqDOut 0) saWait0
1327 -- K^T's fragments for dQ, once dP's products are done
1328 (saWide saValuePointer sqKTOffset saOne (saArgument 3)
1329 (sqLoadKeysT seq
1330 (saMaskTile
1331 (sqScoreGradient
1332 (saPackP
1333 (saValues
1334 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.