def finish

stelliferous@0.0.2.0 · kernel_pipeline.bend:180

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/tensor.bend as Tensors
import stelliferous@0.0.2.0/traversal_partition.bend as Partition

180def finish(partitions: Outputs(),+rows: U32,+positions: U32) -> Storage.Buffer<F32>:
181  match partitions:
182    case Partition.Leaf{input,+offset,+span,channels}:
183      finish_leaf(U32.is_zero(offset) && U32.is_eq(span,positions),input,channels,offset,span,positions)
184    case Partition.Fork{left,right}:
185      gather(Partition.Fork{left,right},Tensors.buffer(~F32,0.0,(rows * positions : U32)),positions)