Skip to content

Commit

Permalink
correcting GEMM_mnk files
Browse files Browse the repository at this point in the history
  • Loading branch information
rraj67 committed Aug 19, 2024
1 parent bff67f3 commit ac60dc9
Show file tree
Hide file tree
Showing 5 changed files with 65 additions and 65 deletions.
26 changes: 13 additions & 13 deletions topologies/GEMM_mnk/NCF.csv
Original file line number Diff line number Diff line change
@@ -1,13 +1,13 @@
Layer,M,N,K
1,256,128,2048
2,128,64,2048
3,256,256,2048
4,2048,256,256
5,2048,256,256
6,2048,128,256
7,2048,256,128
8,2048,128,64
9,2048,64,128
10,128,1,2048
11,2048,1,128
12,2048,128,1
Layer,M,N,K,
1,256,128,2048,
2,128,64,2048,
3,256,256,2048,
4,2048,256,256,
5,2048,256,256,
6,2048,128,256,
7,2048,256,128,
8,2048,128,64,
9,2048,64,128,
10,128,1,2048,
11,2048,1,128,
12,2048,128,1,
36 changes: 18 additions & 18 deletions topologies/GEMM_mnk/gnmt.csv
Original file line number Diff line number Diff line change
@@ -1,18 +1,18 @@
Layer,M,N,K
1,2048,4096,32
2,2048,4096,32
3,2048,1024,32
4,32,2048,4096
5,32,2048,4096
6,32,2048,1024
7,32,3072,4096
8,32,4096,2048
9,32,4096,2048
10,32,4096,3072
11,32,1024,2048
12,3072,4096,32
13,1024,36548,1632
14,1024,1024,1600
15,1632,36548,1024
16,1632,1024,36548
17,1600,1024,1024
Layer,M,N,K,
1,2048,4096,32,
2,2048,4096,32,
3,2048,1024,32,
4,32,2048,4096,
5,32,2048,4096,
6,32,2048,1024,
7,32,3072,4096,
8,32,4096,2048,
9,32,4096,2048,
10,32,4096,3072,
11,32,1024,2048,
12,3072,4096,32,
13,1024,36548,1632,
14,1024,1024,1600,
15,1632,36548,1024,
16,1632,1024,36548,
17,1600,1024,1024,
14 changes: 7 additions & 7 deletions topologies/GEMM_mnk/gpt2.csv
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
Layer,M,N,K
QKT,1024,1024,64
QKTV,1024,64,1024
Linear1,1024,4800,1600
Linear2,1024,1600,1600
PW-FF-L1,1024,3072,1600
PW-FF-L2,1024,1600,3072
Layer,M,N,K,
QKT,1024,1024,64,
QKTV,1024,64,1024,
Linear1,1024,4800,1600,
Linear2,1024,1600,1600,
PW-FF-L1,1024,3072,1600,
PW-FF-L2,1024,1600,3072,
14 changes: 7 additions & 7 deletions topologies/GEMM_mnk/transformer_partial.csv
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
Layer,M,N,K
MH_FC_DimReduce_VKQ_0,128,512,1536
SD_MatMul_QK_00,128,64,128
SD_MatMul_V_00,128,64,128
MH_FC_DimRecast_0,128,512,512
FF_A_0,128,512,2048
FF_B_0,128,2048,2048
Layer,M,N,K,
MH_FC_DimReduce_VKQ_0,128,512,1536,
SD_MatMul_QK_00,128,64,128,
SD_MatMul_V_00,128,64,128,
MH_FC_DimRecast_0,128,512,512,
FF_A_0,128,512,2048,
FF_B_0,128,2048,2048,
40 changes: 20 additions & 20 deletions topologies/GEMM_mnk/unet2d.csv
Original file line number Diff line number Diff line change
@@ -1,20 +1,20 @@
Layer,M,N,K
Conv1a,4186116,64,9
Conv1b,4177936,64,576
Conv2a,1040400,128,576
Conv2b,1036324,128,1152
Conv3a,257049,256,1152
Conv3b,255025,256,2304
Conv4a,63001,512,2304
Conv4b,62001,512,4608
Conv5a,15129,1024,4608
Conv5b,14641,1024,9216
Conv6a,57121,512,9216
Conv6b,56169,512,4608
Conv7a,221841,256,4608
Conv7b,219961,256,2304
Conv8a,876096,128,2304
Conv8b,872356,128,1152
Conv9a,3481956,64,1152
Conv9b,3474496,64,576
Conv10,3474496,2,64
Layer,M,N,K,
Conv1a,4186116,64,9,
Conv1b,4177936,64,576,
Conv2a,1040400,128,576,
Conv2b,1036324,128,1152,
Conv3a,257049,256,1152,
Conv3b,255025,256,2304,
Conv4a,63001,512,2304,
Conv4b,62001,512,4608,
Conv5a,15129,1024,4608,
Conv5b,14641,1024,9216,
Conv6a,57121,512,9216,
Conv6b,56169,512,4608,
Conv7a,221841,256,4608,
Conv7b,219961,256,2304,
Conv8a,876096,128,2304,
Conv8b,872356,128,1152,
Conv9a,3481956,64,1152,
Conv9b,3474496,64,576,
Conv10,3474496,2,64,

0 comments on commit ac60dc9

Please sign in to comment.