Skip to content

Commit

Permalink
extend time limit for matmul_ijk_relu test code and created dot graph…
Browse files Browse the repository at this point in the history
… for matmul_ijk_crddrop_relu
  • Loading branch information
bobcheng15 committed Oct 12, 2023
1 parent 4459d4b commit a10913e
Show file tree
Hide file tree
Showing 3 changed files with 117 additions and 4 deletions.
55 changes: 55 additions & 0 deletions compiler/sam-outputs/onyx-dot/matmul_ijk_crddrop.gv
Original file line number Diff line number Diff line change
@@ -0,0 +1,55 @@
digraph SAM {
comment="X=ss01,B=ss01,C=ss10"
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=broadcast" shape=point style=invis type="broadcast"]
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"]
15 [comment="type=repsiggen,index=i" label="RepeatSignalGenerator i" color=cyan3 shape=box style=filled type="repsiggen" index="i"]
14 [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"]
13 [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"]
12 [comment="type=broadcast" shape=point style=invis type="broadcast"]
1 [comment="type=fiberwrite,index=j,tensor=X,mode=1,format=compressed,segsize=B0_dim+1,crdsize=B0_dim*C1_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*C1_dim" sink="true"]
11 [comment="type=repsiggen,index=j" label="RepeatSignalGenerator j" color=cyan3 shape=box style=filled type="repsiggen" index="j"]
10 [comment="type=repeat,index=j,tensor=B,root=false" label="Repeat j: B" color=cyan2 shape=box style=filled type="repeat" index="j" tensor="B" root="false"]
8 [comment="type=fiberlookup,index=k,tensor=B,mode=1,format=compressed,src=true,root=false" label="FiberLookup k: B1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="k" tensor="B" mode="1" format="compressed" src="true" root="false"]
7 [comment="type=intersect,index=k" label="intersect k" color=purple shape=box style=filled type="intersect" index="k"]
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*C1_dim,sink=true" label="FiberWrite Vals: X" color=green3 shape=box style=filled type="fiberwrite" tensor="X" mode="vals" size="1*B0_dim*C1_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=k,tensor=C,mode=0,format=compressed,src=true,root=false" label="FiberLookup k: C0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="k" tensor="C" mode="0" format="compressed" src="true" root="false"]
18 [comment="type=crddrop,outer=i,inner=j" label="CrdDrop i,j" color=orange shape=box style=filled type="crddrop" outer="i" inner="j"]
19 [comment="type=crddrop,outer=j,inner=k" label="CrdDrop j,k" color=orange shape=box style=filled type="crddrop" outer="j" inner="k"]

17 -> 16 [label="crd" style=dashed type="crd" comment=""]
# 16 -> 2 [label="crd" style=dashed type="crd"]
16 -> 15 [label="crd" style=dashed type="crd"]
15 -> 14 [label="repsig" style=dotted type="repsig"]
14 -> 13 [label="ref" style=bold type="ref"]
13 -> 12 [label="crd" style=dashed type="crd" comment=""]
# 12 -> 1 [label="crd" style=dashed type="crd"]
12 -> 11 [label="crd" style=dashed type="crd"]
11 -> 10 [label="repsig" style=dotted type="repsig"]
10 -> 8 [label="ref" style=bold type="ref"]
8 -> 7 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
7 -> 5 [label="ref_out-B" style=bold type="ref" comment="out-B"]
5 -> 4 [label="val" type="val"]
# 4 -> 3 [label="val" type="val"]
3 -> 0 [label="val" type="val"]
7 -> 6 [label="ref_out-C" style=bold type="ref" comment="out-C"]
6 -> 4 [label="val" type="val"]
8 -> 7 [label="ref_in-B" style=bold type="ref" comment="in-B"]
13 -> 9 [label="ref" style=bold type="ref" comment=""]
9 -> 7 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
9 -> 7 [label="ref_in-C" style=bold type="ref" comment="in-C"]
17 -> 10 [label="ref" style=bold type="ref" comment=""]

4 -> 19 [label="val_inner-k" type="val" comment="inner-k"]
12 -> 19 [label="crd_outer-j" style=dashed type="crd" comment="outer-j"]
18 -> 1 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]
19 -> 3 [label="val_inner-k" type="val" comment="inner-k"]

19 -> 18 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]
16 -> 18 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
18 -> 2 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
}
60 changes: 60 additions & 0 deletions compiler/sam-outputs/onyx-dot/matmul_ijk_crddrop_relu.gv
Original file line number Diff line number Diff line change
@@ -0,0 +1,60 @@
digraph SAM {
comment="X=ss01,B=ss01,C=ss10"
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=broadcast" shape=point style=invis type="broadcast"]
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"]
15 [comment="type=repsiggen,index=i" label="RepeatSignalGenerator i" color=cyan3 shape=box style=filled type="repsiggen" index="i"]
14 [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"]
13 [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"]
12 [comment="type=broadcast" shape=point style=invis type="broadcast"]
1 [comment="type=fiberwrite,index=j,tensor=X,mode=1,format=compressed,segsize=B0_dim+1,crdsize=B0_dim*C1_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*C1_dim" sink="true"]
11 [comment="type=repsiggen,index=j" label="RepeatSignalGenerator j" color=cyan3 shape=box style=filled type="repsiggen" index="j"]
10 [comment="type=repeat,index=j,tensor=B,root=false" label="Repeat j: B" color=cyan2 shape=box style=filled type="repeat" index="j" tensor="B" root="false"]
8 [comment="type=fiberlookup,index=k,tensor=B,mode=1,format=compressed,src=true,root=false" label="FiberLookup k: B1\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="k" tensor="B" mode="1" format="compressed" src="true" root="false"]
7 [comment="type=intersect,index=k" label="intersect k" color=purple shape=box style=filled type="intersect" index="k"]
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*C1_dim,sink=true" label="FiberWrite Vals: X" color=green3 shape=box style=filled type="fiberwrite" tensor="X" mode="vals" size="1*B0_dim*C1_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=k,tensor=C,mode=0,format=compressed,src=true,root=false" label="FiberLookup k: C0\ncompressed" color=green4 shape=box style=filled type="fiberlookup" index="k" tensor="C" mode="0" format="compressed" src="true" root="false"]
18 [comment="type=crddrop,outer=i,inner=j" label="CrdDrop i,j" color=orange shape=box style=filled type="crddrop" outer="i" inner="j"]
19 [comment="type=crddrop,outer=j,inner=k" label="CrdDrop j,k" color=orange shape=box style=filled type="crddrop" outer="j" inner="k"]
20 [comment="type=max" label="Max 0" color=orange shape=box style=filled type="max"]
21 [comment="type=valdrop,outer=j,inner=val" label="ValDropper j,val" color=purple shape=box style=filled type="valdrop" outer="j" inner="val"]
22 [comment="type=crddrop,outer=i,inner=j" label="CrdDrop i,j" color=orange shape=box style=filled type="crddrop" outer="i" inner="j"]
17 -> 16 [label="crd" style=dashed type="crd" comment=""]
16 -> 15 [label="crd" style=dashed type="crd"]
15 -> 14 [label="repsig" style=dotted type="repsig"]
14 -> 13 [label="ref" style=bold type="ref"]
13 -> 12 [label="crd" style=dashed type="crd" comment=""]
12 -> 11 [label="crd" style=dashed type="crd"]
11 -> 10 [label="repsig" style=dotted type="repsig"]
10 -> 8 [label="ref" style=bold type="ref"]
8 -> 7 [label="crd_in-B" style=dashed type="crd" comment="in-B"]
7 -> 5 [label="ref_out-B" style=bold type="ref" comment="out-B"]
5 -> 4 [label="val" type="val"]
7 -> 6 [label="ref_out-C" style=bold type="ref" comment="out-C"]
6 -> 4 [label="val" type="val"]
8 -> 7 [label="ref_in-B" style=bold type="ref" comment="in-B"]
13 -> 9 [label="ref" style=bold type="ref" comment=""]
9 -> 7 [label="crd_in-C" style=dashed type="crd" comment="in-C"]
9 -> 7 [label="ref_in-C" style=bold type="ref" comment="in-C"]
17 -> 10 [label="ref" style=bold type="ref" comment=""]

4 -> 19 [label="val_inner-k" type="val" comment="inner-k"]
12 -> 19 [label="crd_outer-j" style=dashed type="crd" comment="outer-j"]
19 -> 3 [label="val_inner-k" type="val" comment="inner-k"]
3 -> 20 [label="val" type="val" comment="val"]
20 -> 21 [label="val" type="val" comment="relu-val"]
18 -> 21 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]
21 -> 22 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]
18 -> 22 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]

19 -> 18 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]
16 -> 18 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
21 -> 0 [label="val" type="val"]

22 -> 2 [label="crd_outer-i" style=dashed type="crd" comment="outer-i"]
22 -> 1 [label="crd_inner-j" style=dashed type="crd" comment="inner-j"]
}
6 changes: 2 additions & 4 deletions sam/sim/test/apps/test_matmul_ijk_crddrop_relu.py
Original file line number Diff line number Diff line change
Expand Up @@ -18,7 +18,7 @@
import os
import csv
cwd = os.getcwd()
formatted_dir = os.getenv('CUSTOM_FORMATTED_PATH', default=os.path.join(cwd, 'mode-formats'))
formatted_dir = os.getenv('SUITESPARSE_FORMATTED_PATH', default=os.path.join(cwd, 'mode-formats'))


# FIXME: Figureout formats
Expand Down Expand Up @@ -61,8 +61,6 @@ def test_matmul_ijk_crddrop(samBench, ssname, cast, check_gold, debug_sim, repor

C_vals_filename = os.path.join(C_dirname, "tensor_C_mode_vals")
C_vals = read_inputs(C_vals_filename, float)
print(C_vals)


fiberlookup_Bi_17 = CompressedCrdRdScan(crd_arr=B_crd0, seg_arr=B_seg0, debug=debug_sim, statistics=report_stats)
repsiggen_i_15 = RepeatSigGen(debug=debug_sim, statistics=report_stats)
Expand Down Expand Up @@ -92,7 +90,7 @@ def test_matmul_ijk_crddrop(samBench, ssname, cast, check_gold, debug_sim, repor
done = False
time_cnt = 0

while not done and time_cnt < 1000:
while not done and time_cnt < TIMEOUT:
if len(in_ref_B) > 0:
fiberlookup_Bi_17.set_in_ref(in_ref_B.pop(0))
repsiggen_i_15.set_istream(fiberlookup_Bi_17.out_crd())
Expand Down

0 comments on commit a10913e

Please sign in to comment.