def block_k

bend-ml-llama@0.2.1.0 · main.bend:972

import Base
import bend-ml-tensor-array@0.1.19.0/main.bend as TA

972# keep: block with the activations (the tests read them), or block_gen without them; chosen before anything is computed
973def block_k(keep: Bool, +c: Cfg, +d: Nat, lw: LW<c>, -cap: Nat, +t: Nat, kv: KV<cap, t>, -room: {True{} == TA.leb(1n+t, cap) : Bool}, +rt: Rot, +x: List<&2, F32>) -> BR<c, cap, t>:
974  match keep:
975    case True{}:
976      block(c, d, lw, cap, t, kv, room, rt, x)
977    case False{}:
978      block_gen(c, d, lw, cap, t, kv, room, rt, x)