def run

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

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

992# pend is the previous layer's result: its output is the next layer's input
993def run(+keep: Bool, +c: Cfg, +d: Nat, -cap: Nat, +t: Nat, -room: {True{} == TA.leb(1n+t, cap) : Bool}, +rt: Rot, ls: List<&1, LW<c>>, cs: List<&1, KV<cap, t>>, al: List<&1, LW<c>>, ac: List<&1, KV<cap, 1n+t>>, aa: List<&2, Act>, pend: BR<c, cap, t>) -> Run<c, cap, t>:
994  match ls cs pend:
995    case Con{l, lt} Con{k, kt} BR{+y, kv, l0, act}:
996      run(keep, c, d, cap, t, room, rt, lt, kt, l0 <> al, kv <> ac, act <> aa, block_k(keep, c, d, l, cap, t, k, room, rt, y))
997    case _ _ BR{+y, kv, l0, act}:
998      Run{y, List.reverse(&1, LW<c>, l0 <> al), List.reverse(&1, KV<cap, 1n+t>, kv <> ac), List.reverse(&2, Act, act <> aa)}