def step

stelliferous@0.0.2.0 · kernel_gemm.bend:65

import Base
import stelliferous@0.0.2.0/kernel_tile.bend as Tile

65# One ordered k step: read the weight tile, then the input tile, then MAC.
66def step(~upper: Bool,state: Loop) -> Loop:
67  Loop{input,weights,bias,output,+index,context,accumulators} = state
68  Context{+row,+position,+rows,+columns,+input_base,+weight_base} = context
69  with_weights(~upper,input,bias,output,index,Context{row,position,rows,columns,input_base,weight_base},input_base,accumulators,
70    Array.get(Tile.Tile8,weights,(weight_base + index : U32)))