Parallel.balanced: def of bendtorch-core@0.1.0.0
Import: import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel
Lines 166-185 of bendtorch-core@0.1.0.0/src/tensor/parallel.bend: https://jend.leeeet.dev/read.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&start_line=166&end_line=185

import Base
import bendtorch-core@0.1.0.0/src/storage/model.bend as Model

# The parallel build returns a balanced tree of its depth, whatever the
# split. It is the certificate the output storage carries, erased at runtime.
def balanced(~Q: Data, ~P: Data, ~A: Type, ~B: Type, ~prepare: @+q: Q -> P, ~split: A & B -> (A & B) & (A & B),
  ~cell: @+p: P -> @+position: Nat -> A & B -> Bool -> F32 & (A & B),
  +q: Q, +depth: Nat, +splits: Nat, +count: Nat, +base: Nat, state: A & B) ->
  Model.Balanced(F32, depth, build(~Q, ~P, ~A, ~B, ~prepare, ~split, ~cell, q, depth, splits, count, base, state)):
  match depth splits:
    case 1n+inner 1n+rest:
      +high = upper(inner, base)
      fork_balanced(~A, ~B, inner, split(state),
        l => r => build(~Q, ~P, ~A, ~B, ~prepare, ~split, ~cell, q, inner, rest, count, base, (l, r)),
        l => r => build(~Q, ~P, ~A, ~B, ~prepare, ~split, ~cell, q, inner, rest, count, high, (l, r)),
        l => r => balanced(~Q, ~P, ~A, ~B, ~prepare, ~split, ~cell, q, inner, rest, count, base, (l, r)),
        l => r => balanced(~Q, ~P, ~A, ~B, ~prepare, ~split, ~cell, q, inner, rest, count, high, (l, r)))
    case 0n _:
      tree_balanced(~P, ~A, ~B, ~cell, prepare(q), 0n, count, base, state,
        tree_certified(~P, ~A, ~B, ~cell, prepare(q), 0n, count, base, state))
    case 1n+inner 0n:
      tree_balanced(~P, ~A, ~B, ~cell, prepare(q), 1n+inner, count, base, state,
        tree_certified(~P, ~A, ~B, ~cell, prepare(q), 1n+inner, count, base, state))

Uses (11):
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Bool
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=F32
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Nat
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Pair
  https://jend.leeeet.dev/definition.txt?path=Base/base.bend&name=Sigma
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/storage/model.bend&name=Balanced
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&name=build
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&name=fork_balanced
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&name=tree_balanced
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&name=tree_certified
  https://jend.leeeet.dev/definition.txt?path=bendtorch-core@0.1.0.0/src/tensor/parallel.bend&name=upper