proofs/kernel_packing_certificate.bend

stelliferous@0.0.2.0 · 20 lines · 2 definitions

1import Base
2import stelliferous@0.0.2.0/kernel_spatial_packing.bend as Packing
3import stelliferous@0.0.2.0/kernel_tile.bend as Tile
4import stelliferous@0.0.2.0/proofs/kernel_packing_witness.bend as PackingWitness
5import stelliferous@0.0.2.0/proofs/storage_tree_model.bend as Tree
6import stelliferous@0.0.2.0/proofs/storage_certificate.bend as Certificate
7import stelliferous@0.0.2.0/proofs/traversal_witness.bend as TraversalWitness
8import stelliferous@0.0.2.0/proofs/traversal_certificate.bend as TraversalCertificate
9
10def preserves(+depth: Nat,call: Packing.Call,-input: Array<F32>,-output: Array<Tile.Tile8>,input_certificate: Certificate.Certificate<F32,input>,
11  equation: {output == Tree.pack(Tile.Tile8,Certificate.normalized(Tile.Tile8,depth,Tree.reflect(Tile.Tile8,output))) : Array<Tile.Tile8>}) ->
12  TraversalWitness.ResultEvidence<Array<F32>,Tile.Tile8,input,depth,Packing.run(call,input,output)>:
13  Packing.Call{+count,+start,+h,+w,+k,+s,+p,+ow,+inner} = call
14  PackingWitness.pack(depth,count,start,input,output,h,w,k,s,p,ow,inner,Certificate.witness(F32,input,input_certificate),Certificate.witness_at(Tile.Tile8,depth,output,equation))
15
16def certificates(call: Packing.Call,-input: Array<F32>,-output: Array<Tile.Tile8>,input_certificate: Certificate.Certificate<F32,input>,output_certificate: Certificate.Certificate<Tile.Tile8,output>) ->
17  TraversalCertificate.Certificates<Array<F32>,Tile.Tile8,Certificate.family(~F32),Packing.run(call,input,output)>:
18  Certificate.Certificate{+depth,+equation} = input_certificate
19  TraversalCertificate.operation(~Array<F32>,~Tile.Tile8,~Packing.Call,~Certificate.family(~F32),~Packing.run,~preserves,
20    call,input,output,Certificate.Certificate{depth,equation},Certificate.Certificate{depth,equation},Certificate.Certificate{depth,equation},output_certificate)