def q8_cols

bendtorch@1.3.0.0 · dtype.bend:474

import Base

474# the m outputs of row i
475def q8_cols(ml: Nat, +i: U32, +j: U32, +k: Nat, +ku: U32, +mu: U32, st: QS) -> QS:
476  match ml st:
477    case 0n QS{x, q, s, y}:
478      QS{x, q, s, y}
479    case 1n+p QS{x, q, s, y}:
480      q8_cols(p, i, (j + 1 : U32), k, ku, mu, q8_put(((i * mu : U32) + j : U32), j, q8_dot(k, (i * ku : U32), (j * ku : U32), 0.0, Array.get(F32, x, (i * ku : U32)), Array.get(U32, q, U32.shrn((j * ku : U32), 2n))), s, y))