def build

bendtorch-core@0.1.0.0 · src/tensor/elementwise.bend:429

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/tensor/parallel.bend as Parallel

429# The parallel build, `Parallel.build` with the elementwise cell: the top
430# `splits` levels fork, and below the forks a thread builds its subtree in
431# index order. Every cell gets the F32 term the sequential loop writes
432# there, so no output word depends on the split.
433def build(+depth: Nat, +splits: Nat, +operation: Operation, +plan: Shape.BroadcastPlan, +count: Nat, +base: Nat,
434  la: Array<F32>, ra: Array<F32>) -> Array<F32>:
435  Parallel.build(~Request, ~Walk, ~Array<F32>, ~Array<F32>, ~walk, ~split, ~cell_at, Request{operation, plan}, depth, splits, count, base, (la, ra))