def prows

bendtorch@1.3.0.0 · main.bend:2432

import Base

2432# rows i.. of a head's P: r holds rows up to i; left rows more after it. Queries are rows of q (qs apart, the head's
2433# columns from qc), keys rows of k (ks apart, from kc).
2434def prows(left: Nat, +i: Nat, +t: Nat, +hd: Nat, +sc: F32, +qs: U32, +qc: U32, +ks: U32, +kc: U32, r: PR) -> QK:
2435  match left r:
2436    case 0n PR{q, k, p, n}:
2437      QK{q, k, p}
2438    case 1n+m PR{q, k, p, n}:
2439      prows(m, 1n+i, t, hd, sc, qs, qc, ks, kc, prow(1n+i, t, hd, sc, qs, qc, ks, kc, q, k, p))