def block_packed

stelliferous@0.0.2.0 · kernel_pipeline.bend:126

import Base
import stelliferous@0.0.2.0/kernel_gemm_buffer.bend as GemmBuffer
import stelliferous@0.0.2.0/kernel_tile.bend as Tile
import stelliferous@0.0.2.0/storage_buffer.bend as Storage

126def block_packed(+rows: U32,+positions: U32,inner: U32,+start: U32,+span: U32,
127  weights: Storage.Buffer<Tile.Tile8>,bias: Storage.Buffer<F32>,output: Storage.Buffer<F32>,packed: Storage.Buffer<F32> & Storage.Buffer<Tile.Tile8>) -> BlockState:
128  (input,tiles) = packed
129  block_kept(input,
130    GemmBuffer.run_keeping(rows,tiles,weights,bias,output,span,inner,start,positions))