def writes

bendtorch-core@0.1.0.0 · src/tensor/contracts/matmul_parallel_proof.bend:40

import Base
import bendtorch-core@0.1.0.0/src/shape/core.bend as Shape
import bendtorch-core@0.1.0.0/src/storage/LAWS.bend as StorageLaws
import bendtorch-core@0.1.0.0/src/tensor/contracts/replayed.bend as Replayed

40def writes(+inner: Nat, +columns: Nat, +left: Shape.Shape, +left_fill: F32, +left_ops: List<&2, StorageLaws.Op>,
41  +right: Shape.Shape, +right_fill: F32, +right_ops: List<&2, StorageLaws.Op>, +count: Nat) -> List<&2, StorageLaws.Op>:
42  match count:
43    case 0n:
44      Nil{}
45    case 1n+previous:
46      Replayed.snoc(writes(inner, columns, left, left_fill, left_ops, right, right_fill, right_ops, previous),
47        StorageLaws.Write{U32.from_nat(previous), cell_word(inner, columns, left, left_fill, left_ops, right, right_fill, right_ops, previous)})