type EAC
import bendtorch-gpu@1.3.0.0/main.bend as BendtorchGpuimport Base
619# the sizes; n is how many cells (or rows) the bang has, nw and nk the cells of dQ and of dK, tr = t(t + 1)/2 the cells
620# of a head's P
621type EAC is Data:
622 EAC{n: U32, t: Nat, t32: U32, h: U32, g: U32, gn: Nat, hd: Nat, hd32: U32, w: U32, kvw: U32, sc: F32, nw: U32, nk: U32, tr: U32}