Skip to content

Commit

Permalink
initial commit suitesparse onyx flow
Browse files Browse the repository at this point in the history
  • Loading branch information
kalhankoul96 committed Jan 4, 2024
1 parent ef9e665 commit a9374b7
Show file tree
Hide file tree
Showing 14 changed files with 654 additions and 3 deletions.
27 changes: 27 additions & 0 deletions compiler/sam-outputs/onyx-dot/masked_broadcast.gv
Original file line number Diff line number Diff line change
@@ -0,0 +1,27 @@
digraph SAM {
comment="X=ss01,B=ss01,c=s0"
0 [comment="type=fiberlookup,index=i,tensor=B,mode=0,format=compressed,src=true,root=true" label="FiberLookup i: B\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="i" tensor="B" mode="0" format="compressed" src="true" root="true"]
1 [comment="type=repsiggen,index=i" label="RepeatSignalGenerator i" color=cyan3 shape=box style=filled type="repsiggen" index="i"]
2 [comment="type=repeat,index=i,tensor=c,root=true" label="Repeat i: B" color=cyan2 shape=box style=filled type="repeat" index="i" tensor="c" root="true"]
3 [comment="type=fiberlookup,index=i,tensor=c,mode=0,format=compressed,src=true,root=false" label="FiberLookup i: c\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="i" tensor="c" mode="0" format="compressed" src="true" root="false"]
4 [comment="type=fiberlookup,index=j,tensor=B,mode=1,format=compressed,src=true,root=false" label="FiberLookup j: B\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="B" mode="1" format="compressed" src="true" root="false"]
5 [comment="type=intersect,index=j" label="intersect j" color=purple shape=box style=filled type="intersect" index="j"]
6 [comment="type=arrayvals,tensor=c" label="Array Vals: c" color=green2 shape=box style=filled type="arrayvals" tensor="c"]
7 [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"]
8 [comment="type=fiberwrite,index=j,tensor=X,mode=1,format=compressed,segsize=B0_dim+1,crdsize=B0_dim*B1_dim,sink=true" label="FiberWrite j: X1\ncompressed" color=green3 shape=box style=filled type="fiberwrite" index="j" tensor="X" mode="1" format="compressed" segsize="B0_dim+1" crdsize="B0_dim*B1_dim" sink="true"]
9 [comment="type=fiberwrite,mode=vals,tensor=X,size=1*B0_dim*B1_dim,sink=true" label="FiberWrite Vals: X" color=green3 shape=box style=filled type="fiberwrite" tensor="X" mode="vals" size="1*B0_dim*B1_dim" sink="true"]
10 [comment="type=broadcast" shape=point style=invis type="broadcast"]
0 -> 10 [label="crd" style=dashed type="crd"]
10 -> 1 [label="crd" style=dashed type="crd"]
1 -> 2 [label="repsig" style=dotted type="repsig"]
2 -> 3 [label="ref" style=bold type="ref"]
0 -> 4 [label="ref" style=bold type="ref"]
3 -> 5 [label="ref_in-c" style=bold type="ref" comment="in-c"]
3 -> 5 [label="crd_in-c" style=dashed type="crd" comment="in-c"]
4 -> 5 [label="ref_in-B", style=bold type="ref" comment="in-B"]
4 -> 5 [label="crd_in-B", style=dashed type="crd" comment="in-B"]
5 -> 6 [label="ref_out-c", style=bold type="ref" comment="out-c"]
10 -> 7 [label="crd", style=dashed type="crd"]
5 -> 8 [label="crd", style=dashed type="crd"]
6 -> 9 [label="val", type="val"]
}
42 changes: 42 additions & 0 deletions compiler/sam-outputs/onyx-dot/mat_elemadd_relu.gv
Original file line number Diff line number Diff line change
@@ -0,0 +1,42 @@
digraph SAM {
comment="X=ss01,B=ss01,C=ss01"
10 [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"]
9 [comment="type=union,index=i" label="union i" color=purple shape=box style=filled type="union" index="i"]
7 [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"]
6 [comment="type=union,index=j" label="union j" color=purple shape=box style=filled type="union" index="j"]
4 [comment="type=arrayvals,tensor=B" label="Array Vals: B" color=green2 shape=box style=filled type="arrayvals" tensor="B"]
3 [comment="type=add,sub=0" label="Add" color=brown shape=box style=filled type="add" sub="0"]
5 [comment="type=arrayvals,tensor=C" label="Array Vals: C" color=green2 shape=box style=filled type="arrayvals" tensor="C"]
8 [comment="type=fiberlookup,index=j,tensor=C,mode=1,format=compressed,src=true,root=false" label="FiberLookup j: C1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="C" mode="1" format="compressed" src="true" root="false"]
11 [comment="type=fiberlookup,index=i,tensor=C,mode=0,format=compressed,src=true,root=true" label="FiberLookup i: C0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="i" tensor="C" mode="0" format="compressed" src="true" root="true"]
12 [comment="type=max" label="Max 0" color=brown shape=box style=filled type="max"]
13 [comment="type=crddrop,outer=j,inner=val,mode=0" label="CrdDrop Compression j, val" color=orange style=filled type="crddrop" outer="j" inner="val" mode="0"]
0 [comment="type=fiberwrite,mode=vals,tensor=X,size=1*B0_dim*B1_dim,sink=true" label="FiberWrite Vals: X" color=green3 shape=box style=filled type="fiberwrite" tensor="X" mode="vals" size="2*B0_dim*B1_dim" sink="true"]
14 [comment="type=crddrop,outer=i,inner=j" label="CrdDrop i,j" color=orange shape=box style=filled type="crddrop" outer="i" inner="j"]
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"]
1 [comment="type=fiberwrite,index=j,tensor=X,mode=1,format=compressed,segsize=B0_dim+1,crdsize=B0_dim*B1_dim,sink=true" label="FiberWrite j: X1\ncompressed" color=green3 shape=box style=filled type="fiberwrite" index="j" tensor="X" mode="1" format="compressed" segsize="B0_dim+1" crdsize="B0_dim*B1_dim" sink="true"]
10 -> 9 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
9 -> 7 [label="ref_out-B" style=bold type="ref" comment="out-B"]
7 -> 6 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
6 -> 4 [label="ref_out-B" style=bold type="ref" comment="out-B"]
4 -> 3 [label="val" type="val"]
6 -> 5 [label="ref_out-C" style=bold type="ref" comment="out-C"]
5 -> 3 [label="val" type="val"]
7 -> 6 [label="ref_in-B" style=bold type="ref" comment="in-B"]
9 -> 8 [label="ref_out-C" style=bold type="ref" comment="out-C"]
8 -> 6 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
8 -> 6 [label="ref_in-C" style=bold type="ref" comment="in-C"]
10 -> 9 [label="ref_in-B" style=bold type="ref" comment="in-B"]
11 -> 9 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
11 -> 9 [label="ref_in-C" style=bold type="ref" comment="in-C"]

3 -> 12 [label="val" type="val" comment="val"]
12 -> 13 [label="val" type="val" comment="inner-val"]
6 -> 13 [label="crd_outer-j" style=dashed type="crd" comment="outer-j"]
13 -> 0 [label="val" type="val", comment="val"]
13 -> 14 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]
9 -> 14 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
14 -> 2 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
14 -> 1 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]

}
62 changes: 62 additions & 0 deletions compiler/sam-outputs/onyx-dot/mat_mask_tri.gv
Original file line number Diff line number Diff line change
@@ -0,0 +1,62 @@
digraph SAM {
comment="x=none,B=ss01,C=ss01,D=ss10"
22 [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"]
21 [comment="type=intersect,index=i" label="intersect i" color=purple shape=box style=filled type="intersect" index="i"]
20 [comment="type=repsiggen,index=i" label="RepeatSignalGenerator i" color=cyan3 shape=box style=filled type="repsiggen" index="i"]
19 [comment="type=repeat,index=i,tensor=D,root=true" label="Repeat i: D" color=cyan2 shape=box style=filled type="repeat" index="i" tensor="D" root="true"]
18 [comment="type=fiberlookup,index=j,tensor=D,mode=1,format=compressed,src=true,root=false" label="FiberLookup j: D1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="D" mode="1" format="compressed" src="true" root="false"]
16 [comment="type=intersect,index=j" label="intersect j" color=purple shape=box style=filled type="intersect" index="j"]
15 [comment="type=repsiggen,index=j" label="RepeatSignalGenerator j" color=cyan3 shape=box style=filled type="repsiggen" index="j"]
14 [comment="type=repeat,index=j,tensor=C,root=false" label="Repeat j: C" color=cyan2 shape=box style=filled type="repeat" index="j" tensor="C" root="false"]
12 [comment="type=fiberlookup,index=k,tensor=C,mode=1,format=compressed,src=true,root=false" label="FiberLookup k: C1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="k" tensor="C" mode="1" format="compressed" src="true" root="false"]
11 [comment="type=intersect,index=k" label="intersect k" color=purple shape=box style=filled type="intersect" index="k"]
10 [comment="type=repsiggen,index=k" label="RepeatSignalGenerator k" color=cyan3 shape=box style=filled type="repsiggen" index="k"]
9 [comment="type=repeat,index=k,tensor=B,root=false" label="Repeat k: B" color=cyan2 shape=box style=filled type="repeat" index="k" tensor="B" root="false"]
6 [comment="type=arrayvals,tensor=B" label="Array Vals: B" color=green2 shape=box style=filled type="arrayvals" tensor="B"]
5 [comment="type=mul" label="Mul" color=brown shape=box style=filled type="mul"]
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"]
2 [comment="type=reduce" label="Reduce" color=brown shape=box style=filled type="reduce"]
1 [comment="type=reduce" label="Reduce" color=brown shape=box style=filled type="reduce"]
0 [comment="type=fiberwrite,mode=vals,tensor=x,size=1,sink=true" label="FiberWrite Vals: x" color=green3 shape=box style=filled type="fiberwrite" tensor="x" mode="vals" size="1" sink="true"]
7 [comment="type=arrayvals,tensor=C" label="Array Vals: C" color=green2 shape=box style=filled type="arrayvals" tensor="C"]
8 [comment="type=arrayvals,tensor=D" label="Array Vals: D" color=green2 shape=box style=filled type="arrayvals" tensor="D"]
13 [comment="type=fiberlookup,index=k,tensor=D,mode=0,format=compressed,src=true,root=false" label="FiberLookup k: D0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="k" tensor="D" mode="0" format="compressed" src="true" root="false"]
17 [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"]
23 [comment="type=fiberlookup,index=i,tensor=C,mode=0,format=compressed,src=true,root=true" label="FiberLookup i: C0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="i" tensor="C" mode="0" format="compressed" src="true" root="true"]
22 -> 21 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
21 -> 20 [label="crd" style=dashed type="crd"]
20 -> 19 [label="repsig" style=dotted type="repsig"]
19 -> 18 [label="ref" style=bold type="ref"]
18 -> 16 [label="crd_in-D" style=dashed type="crd" comment="in-D"]
16 -> 15 [label="crd" style=dashed type="crd"]
15 -> 14 [label="repsig" style=dotted type="repsig"]
14 -> 12 [label="ref" style=bold type="ref"]
12 -> 11 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
11 -> 10 [label="crd" style=dashed type="crd"]
10 -> 9 [label="repsig" style=dotted type="repsig"]
9 -> 6 [label="ref" style=bold type="ref"]
6 -> 5 [label="val" type="val"]
5 -> 4 [label="val" type="val"]
4 -> 3 [label="val" type="val"]
3 -> 2 [label="val" type="val"]
2 -> 1 [label="val" type="val"]
1 -> 0 [label="val" type="val"]
11 -> 7 [label="ref_out-C" style=bold type="ref" comment="out-C"]
7 -> 5 [label="val" type="val"]
11 -> 8 [label="ref_out-D" style=bold type="ref" comment="out-D"]
8 -> 4 [label="val" type="val"]
12 -> 11 [label="ref_in-C" style=bold type="ref" comment="in-C"]
16 -> 9 [label="ref_out-B" style=bold type="ref" comment="out-B"]
16 -> 13 [label="ref_out-D" style=bold type="ref" comment="out-D"]
13 -> 11 [label="crd_in-D" style=dashed type="crd" comment="in-D"]
13 -> 11 [label="ref_in-D" style=bold type="ref" comment="in-D"]
18 -> 16 [label="ref_in-D" style=bold type="ref" comment="in-D"]
21 -> 17 [label="ref_out-B" style=bold type="ref" comment="out-B"]
17 -> 16 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
17 -> 16 [label="ref_in-B" style=bold type="ref" comment="in-B"]
21 -> 14 [label="ref_out-C" style=bold type="ref" comment="out-C"]
22 -> 21 [label="ref_in-B" style=bold type="ref" comment="in-B"]
23 -> 21 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
23 -> 21 [label="ref_in-C" style=bold type="ref" comment="in-C"]
}
47 changes: 47 additions & 0 deletions compiler/sam-outputs/onyx-dot/mat_residual.gv
Original file line number Diff line number Diff line change
@@ -0,0 +1,47 @@
digraph SAM {
comment="x=s0,b=s0,C=ss01,d=s0"
17 [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"]
16 [comment="type=union,index=i" label="union i" color=purple shape=box style=filled type="union" index="i"]
15 [comment="type=broadcast" shape=point style=invis type="broadcast"]
9 [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"]
14 [comment="type=repsiggen,index=i" label="RepeatSignalGenerator i" color=cyan3 shape=box style=filled type="repsiggen" index="i"]
13 [comment="type=repeat,index=i,tensor=d,root=true" label="Repeat i: d" color=cyan2 shape=box style=filled type="repeat" index="i" tensor="d" root="true"]
12 [comment="type=fiberlookup,index=j,tensor=d,mode=0,format=compressed,src=true,root=false" label="FiberLookup j: d0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="d" mode="0" format="compressed" src="true" root="false"]
10 [comment="type=intersect,index=j" label="intersect j" color=purple shape=box style=filled type="intersect" index="j"]
7 [comment="type=arrayvals,tensor=C" label="Array Vals: C" color=green2 shape=box style=filled type="arrayvals" tensor="C"]
6 [comment="type=mul" label="Mul" color=brown shape=box style=filled type="mul"]
5 [comment="type=reduce" label="Reduce" color=brown shape=box style=filled type="reduce"]
3 [comment="type=add,sub=1" label="Add
subtract" color=brown shape=box style=filled type="add" sub="1"]
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"]
8 [comment="type=arrayvals,tensor=d" label="Array Vals: d" color=green2 shape=box style=filled type="arrayvals" tensor="d"]
4 [comment="type=arrayvals,tensor=b" label="Array Vals: b" color=green2 shape=box style=filled type="arrayvals" tensor="b"]
11 [comment="type=fiberlookup,index=j,tensor=C,mode=1,format=compressed,src=true,root=false" label="FiberLookup j: C1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="j" tensor="C" mode="1" format="compressed" src="true" root="false"]
18 [comment="type=fiberlookup,index=i,tensor=C,mode=0,format=compressed,src=true,root=true" label="FiberLookup i: C0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="i" tensor="C" mode="0" format="compressed" src="true" root="true"]
17 -> 16 [label="crd_in-b" style=dashed type="crd" comment="in-b"]
16 -> 15 [label="crd_i" style=dashed type="crd" comment="i"]
15 -> 9 [label="crd_i" style=dashed type="crd" comment="i"]
9 -> 1 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
15 -> 14 [label="crd" style=dashed type="crd" comment=""]
14 -> 13 [label="repsig" style=dotted type="repsig"]
13 -> 12 [label="ref" style=bold type="ref"]
12 -> 10 [label="crd_in-d" style=dashed type="crd" comment="in-d"]
10 -> 9 [label="crd_in-j" style=dashed type="crd" comment="in-j"]
10 -> 7 [label="ref_out-C" style=bold type="ref" comment="out-C"]
7 -> 6 [label="val" type="val"]
6 -> 5 [label="val" type="val"]
5 -> 3 [label="val" type="val"]
3 -> 0 [label="val" type="val"]
10 -> 8 [label="ref_out-d" style=bold type="ref" comment="out-d"]
8 -> 6 [label="val" type="val"]
12 -> 10 [label="ref_in-d" style=bold type="ref" comment="in-d"]
16 -> 4 [label="ref_out-b" style=bold type="ref" comment="out-b"]
4 -> 3 [label="val" type="val"]
16 -> 11 [label="ref_out-C" style=bold type="ref" comment="out-C"]
11 -> 10 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
11 -> 10 [label="ref_in-C" style=bold type="ref" comment="in-C"]
17 -> 16 [label="ref_in-b" style=bold type="ref" comment="in-b"]
18 -> 16 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
18 -> 16 [label="ref_in-C" style=bold type="ref" comment="in-C"]
}
7 changes: 6 additions & 1 deletion compiler/sam-outputs/onyx-dot/mat_vecmul_ij.gv
Original file line number Diff line number Diff line change
Expand Up @@ -3,6 +3,7 @@ digraph SAM {
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"]
15 [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"]
Expand All @@ -24,12 +25,16 @@ digraph SAM {
8 -> 7 [label="crd_in-j" style=dashed type="crd" comment="in-j"]
8 -> 5 [label="ref_out-B" style=bold type="ref" comment="out-B"]
5 -> 4 [label="val" type="val"]
4 -> 3 [label="val" type="val"]
# 4 -> 3 [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"]

4 -> 15 [label="val_inner-j" type="val" comment="inner-j"]
15 -> 3 [label="val_inner-j" type="val" comment="inner-j"]
13 -> 15 [label="crd_i" style=dashed type="crd" comment="i"]
}
Loading

0 comments on commit a9374b7

Please sign in to comment.