Skip to content

Commit

Permalink
added graph for spmv and spmv_relu
Browse files Browse the repository at this point in the history
  • Loading branch information
@bobcheng15
bobcheng15 committed Dec 7, 2023
1 parent 279a49c commit c82d99e
Show file tree
Hide file tree
Showing 2 changed files with 84 additions and 0 deletions.
37 changes: 37 additions & 0 deletions compiler/sam-outputs/onyx-dot/spmv.gv
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,37 @@
digraph SAM {
comment="x=s0,B=ss01,c=d0"
14 [comment="type=fiberlookup,index=i,tensor=B,mode=0,format=compressed,src=true,root=true" label="FiberLookup i: B0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="i" tensor="B" mode="0" format="compressed" src="true" root="true"]
13 [comment="type=broadcast" shape=point style=invis type="broadcast"]
7 [comment="type=crddrop,outer=i,inner=j" label="CrdDrop i,j" color=orange shape=box style=filled type="crddrop" outer="i" inner="j"]
1 [comment="type=fiberwrite,index=i,tensor=x,mode=0,format=compressed,segsize=2,crdsize=B0_dim,sink=true" label="FiberWrite i: x0\ncompressed" color=green3 shape=box style=filled type="fiberwrite" index="i" tensor="x" mode="0" format="compressed" segsize="2" crdsize="B0_dim" sink="true"]
12 [comment="type=repsiggen,index=i" label="RepeatSignalGenerator i" color=cyan3 shape=box style=filled type="repsiggen" index="i"]
11 [comment="type=repeat,index=i,tensor=c,root=true" label="Repeat i: c" color=cyan2 shape=box style=filled type="repeat" index="i" tensor="c" root="true"]
10 [comment="type=fiberlookup,index=j,tensor=c,mode=0,format=dense,src=true,root=false" label="FiberLookup j: c0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="c" mode="0" format="dense" src="true" root="false"]
8 [comment="type=intersect,index=j" label="intersect j" color=purple shape=box style=filled type="intersect" index="j"]
5 [comment="type=arrayvals,tensor=B" label="Array Vals: B" color=green2 shape=box style=filled type="arrayvals" tensor="B"]
4 [comment="type=mul" label="Mul" color=brown shape=box style=filled type="mul"]
3 [comment="type=reduce" label="Reduce" color=brown shape=box style=filled type="reduce"]
0 [comment="type=fiberwrite,mode=vals,tensor=x,size=1*B0_dim,sink=true" label="FiberWrite Vals: x" color=green3 shape=box style=filled type="fiberwrite" tensor="x" mode="vals" size="1*B0_dim" sink="true"]
6 [comment="type=arrayvals,tensor=c" label="Array Vals: c" color=green2 shape=box style=filled type="arrayvals" tensor="c"]
9 [comment="type=fiberlookup,index=j,tensor=B,mode=1,format=compressed,src=true,root=false" label="FiberLookup j: B1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="B" mode="1" format="compressed" src="true" root="false"]
14 -> 13 [label="crd" style=dashed type="crd" comment=""]
13 -> 7 [label="crd_i" style=dashed type="crd" comment="i"]
7 -> 1 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
13 -> 12 [label="crd" style=dashed type="crd" comment=""]
12 -> 11 [label="repsig" style=dotted type="repsig"]
11 -> 10 [label="ref" style=bold type="ref"]
10 -> 8 [label="crd_in-c" style=dashed type="crd" comment="in-c"]
8 -> 5 [label="ref_out-B" style=bold type="ref" comment="out-B"]
5 -> 4 [label="val" type="val"]
3 -> 0 [label="val" type="val"]
8 -> 6 [label="ref_out-c" style=bold type="ref" comment="out-c"]
6 -> 4 [label="val" type="val"]
10 -> 8 [label="ref_in-c" style=bold type="ref" comment="in-c"]
14 -> 9 [label="ref" style=bold type="ref" comment=""]
9 -> 8 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
9 -> 8 [label="ref_in-B" style=bold type="ref" comment="in-B"]

8 -> 7 [label="crd_in-j" style=dashed type="crd" comment="in-j"]
4 -> 3 [label="val" type="val"]

}
47 changes: 47 additions & 0 deletions compiler/sam-outputs/onyx-dot/spmv_relu.gv
View file
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
digraph SAM {
comment="x=s0,B=ss01,c=d0"
14 [comment="type=fiberlookup,index=i,tensor=B,mode=0,format=compressed,src=true,root=true" label="FiberLookup i: B0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="i" tensor="B" mode="0" format="compressed" src="true" root="true"]
13 [comment="type=broadcast" shape=point style=invis type="broadcast"]
12 [comment="type=repsiggen,index=i" label="RepeatSignalGenerator i" color=cyan3 shape=box style=filled type="repsiggen" index="i"]
11 [comment="type=repeat,index=i,tensor=c,root=true" label="Repeat i: c" color=cyan2 shape=box style=filled type="repeat" index="i" tensor="c" root="true"]
10 [comment="type=fiberlookup,index=j,tensor=c,mode=0,format=dense,src=true,root=false" label="FiberLookup j: c0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="c" mode="0" format="dense" src="true" root="false"]
8 [comment="type=intersect,index=j" label="intersect j" color=purple shape=box style=filled type="intersect" index="j"]
5 [comment="type=arrayvals,tensor=B" label="Array Vals: B" color=green2 shape=box style=filled type="arrayvals" tensor="B"]
4 [comment="type=mul" label="Mul" color=brown shape=box style=filled type="mul"]
3 [comment="type=reduce" label="Reduce" color=brown shape=box style=filled type="reduce"]
6 [comment="type=arrayvals,tensor=c" label="Array Vals: c" color=green2 shape=box style=filled type="arrayvals" tensor="c"]
9 [comment="type=fiberlookup,index=j,tensor=B,mode=1,format=compressed,src=true,root=false" label="FiberLookup j: B1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="B" mode="1" format="compressed" src="true" root="false"]


20 [comment="type=max" label="Max 0" color=brown shape=box style=filled type="max"]
0 [comment="type=fiberwrite,mode=vals,tensor=x,size=1*B0_dim,sink=true" label="FiberWrite Vals: x" color=green3 shape=box style=filled type="fiberwrite" tensor="x" mode="vals" size="1*B0_dim" sink="true"]
21 [comment="type=crddrop,outer=i,inner=val,mode=0" label="CrdDrop Compression i, val" color=orange style=filled type="crddrop" outer="i" inner="val" mode="0"]
2 [comment="type=fiberwrite,index=i,tensor=x,mode=0,format=compressed,segsize=2,crdsize=B0_dim,sink=true" label="FiberWrite i: x0\ncompressed" color=green3 shape=box style=filled type="fiberwrite" index="i" tensor="x" mode="0" format="compressed" segsize="2" crdsize="B0_dim" sink="true"]



14 -> 13 [label="crd" style=dashed type="crd" comment=""]
13 -> 12 [label="crd" style=dashed type="crd" comment=""]
12 -> 11 [label="repsig" style=dotted type="repsig"]
11 -> 10 [label="ref" style=bold type="ref"]
10 -> 8 [label="crd_in-c" style=dashed type="crd" comment="in-c"]
8 -> 5 [label="ref_out-B" style=bold type="ref" comment="out-B"]
5 -> 4 [label="val" type="val"]
8 -> 6 [label="ref_out-c" style=bold type="ref" comment="out-c"]
6 -> 4 [label="val" type="val"]
10 -> 8 [label="ref_in-c" style=bold type="ref" comment="in-c"]
14 -> 9 [label="ref" style=bold type="ref" comment=""]
9 -> 8 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
9 -> 8 [label="ref_in-B" style=bold type="ref" comment="in-B"]


4 -> 3 [label="val" type="val"]


3 -> 20 [label="val" type="val" comment="val"]
20 -> 21 [label="val" type="val" comment="inner-val"]
13 -> 21 [label="crd_i" style=dashed type="crd" comment="i"]
21 -> 0 [label="val" type="val", comment="val"]
21 -> 2 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]

}

0 comments on commit c82d99e

Please sign in to comment.