def finish_leaf

stelliferous@0.0.2.0 · kernel_pipeline.bend:175

import Base
import stelliferous@0.0.2.0/kernel_gather_buffer.bend as GatherBuffer
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/tensor.bend as Tensors

175def finish_leaf(identity: Bool,input: Storage.Buffer<F32>,+rows: U32,offset: U32,span: U32,+positions: U32) -> Storage.Buffer<F32>:
176  match identity:
177    case True{}: input
178    case False{}: GatherBuffer.run(rows,offset,positions,span,input,Tensors.buffer(~F32,0.0,(rows * positions : U32)))