1 import Base 2 import ../tensor_cursor.bend as Cursor 3 import ./storage_witness.bend as Witness 4 import ./traversal_visit_witness.bend as VisitWitness 5 6 def read_value(~Element: Data,axes: Cursor.Levels(),address: U32,-storage: Array,-result: Array & Element, 7 observed: Witness.Read) -> 8 VisitWitness.Sample,Element,Cursor.Cursor,storage,Cursor.read_value(~Element,axes,address,result)>: 9 match observed: 10 case Witness.Read{value,equation}: 11 %Equal.sym(Array & Element,result,(storage,value),equation) : 12 VisitWitness.Sample,Element,Cursor.Cursor,storage,Cursor.read_value(~Element,axes,address,_)> 13 VisitWitness.Sample{Cursor.advance(axes,address),value,{==}} 14 15 def read(~Element: Data,-storage: Array,cursor: Cursor.Cursor,evidence: Witness.ArrayWitness) -> 16 VisitWitness.Sample,Element,Cursor.Cursor,storage,Cursor.read(~Element,storage,cursor)>: 17 Cursor.Cursor{axes,+address} = cursor 18 read_value(~Element,axes,address,storage,Array.get(Element,storage,address),Witness.observed(Element,storage,address,evidence)) 19 20 def stride_value(~Element: Data,address: U32,+stride: U32,-storage: Array,-result: Array & Element, 21 observed: Witness.Read) -> 22 VisitWitness.Sample,Element,Cursor.Stride,storage,Cursor.stride_value(~Element,address,stride,result)>: 23 match observed: 24 case Witness.Read{value,equation}: 25 %Equal.sym(Array & Element,result,(storage,value),equation) : 26 VisitWitness.Sample,Element,Cursor.Stride,storage,Cursor.stride_value(~Element,address,stride,_)> 27 VisitWitness.Sample{Cursor.Stride{(address + stride : U32),stride},value,{==}} 28 29 def read_stride(~Element: Data,-storage: Array,cursor: Cursor.Stride,evidence: Witness.ArrayWitness) -> 30 VisitWitness.Sample,Element,Cursor.Stride,storage,Cursor.read_stride(~Element,storage,cursor)>: 31 Cursor.Stride{+address,+stride} = cursor 32 stride_value(~Element,address,stride,storage,Array.get(Element,storage,address),Witness.observed(Element,storage,address,evidence))