def pair

bendtorch-core@0.1.0.0 · src/tensor/parallel.bend:44

import Base

44def pair(~P: Data, ~A: Type, ~B: Type, ~cell: @+p: P -> @+position: Nat -> A & B -> Bool -> F32 & (A & B),
45  +p: P, +count: Nat, +base: Nat, low: F32 & (A & B)) -> Array<F32> & (A & B):
46  (x, state) = low
47  (l, r) = state
48  pair_join(~(A & B), x, cell(p, Nat.add(1n, base), (l, r), Nat.is_lt(Nat.add(1n, base), count)))