def reorder

stelliferous@0.0.2.0 · kernel_gemm_buffer.bend:39

import Base
import stelliferous@0.0.2.0/kernel_tile.bend as Tile
import stelliferous@0.0.2.0/storage_buffer.bend as Storage
import stelliferous@0.0.2.0/kernel_weights.bend as Weights
import stelliferous@0.0.2.0/proofs/kernel_weights_certificate.bend as WeightCertificate

39def reorder(+rows: U32,+inner: U32,weights: Storage.Buffer<F32>) -> Storage.Buffer<Tile.Tile8>:
40  Storage.Buffer{weights,length,certificate} = weights
41  Storage.Buffer{Weights.run(rows,inner,weights),U32.to_nat((((rows + 7) / 8) * inner : U32)),
42    WeightCertificate.certificate(rows,inner,weights,certificate)}