def pair_when

stelliferous@0.0.2.0 · tensor_lanes.bend:71

import Base
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/tensor_operations.bend as Operations
import stelliferous@0.0.2.0/tensor_view.bend as Views

71def pair_when(~operation: F32 -> F32 -> F32,fast: Bool,count: Nat,right_view: Views.View,offset: U32,
72  left: Storage.Buffer<F32>,right: Storage.Buffer<F32>) -> Storage.Buffer<F32> & Storage.Buffer<F32>:
73  match fast:
74    case True{}: pair_buffers(pair_map(~operation,count,0,left,right))
75    case False{}: Operations.combine_kernel(~F32,~operation,count,right_view,offset,left,right)