def join_compatible_selected

stelliferous@0.0.2.0 · tensor_layout.bend:346

import Base
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/tensor_band.bend as Bands

346# The final branch is one Bool. Plan comparison stays in the metadata pass,
347# outside every element-copy loop. Dense append gathers each band source once.
348def join_compatible_selected(~Element: Data,compatible: Bool,+shape: List<&2,U32>,+axis: U32,+fill: Element,+workers: U32,
349  discovered: Tensors.Parts<Element> & Maybe<&2,Bands.Plan>) -> Tensors.Tensor<Element>:
350  match compatible:
351    case True{}: join_discovered(~Element,shape,axis,fill,workers,discovered)
352    case False{}:
353      (parts,layout) = discovered
354      append_parts(~Element,parts,axis,(Tensors.with_workers(~Element,Tensors.full(~Element,shape,fill),workers),0))