add batch_count=64 Exact sizes for NN sgemm
@@ -436,9 +436,6 @@ BenchmarkProblems:
- KernelLanguage: ["Assembly"]
- PrefetchGlobalRead: [ True ]
- PrefetchLocalRead: [ True ]
-# - PrefetchGlobalRead: [0,1]
-# - PrefetchLocalRead: [0,1]
-# - WorkGroupMapping: [1, 8]
ForkParameters:
- ThreadTile:
- [ 2, 2 ]
@@ -449,11 +446,6 @@ BenchmarkProblems:
- [ 8, 4 ]
- [ 8, 8 ]
- WorkGroup:
-# - [ 4, 16, 1 ]
-# - [ 16, 4, 1 ]
-# - [ 4, 32, 1 ]
-# - [ 32, 4, 1 ]
-# - [ 8, 32, 1 ]
- [ 8, 8, 1 ]
- [ 16, 8, 1 ]
- [ 16, 16, 1 ]
@@ -479,6 +471,52 @@ BenchmarkProblems:
- Exact: [ 2048, 3136, 1, 512 ]
- Exact: [ 512, 3136, 1, 2048 ]
+ ########################################
+ # NN - Batch
+ ########################################
+ - # Benchmark Group
+ InitialSolutionParameters:
+ BenchmarkCommonParameters:
+ - EdgeType: ["ShiftPtr"]
+ - LoopTail: [True]
+ - KernelLanguage: ["Assembly"]
+ - PrefetchGlobalRead: [ True ]
+ - PrefetchLocalRead: [ True ]
+ ForkParameters:
+ - ThreadTile:
+# - [ 2, 2 ]
+ - [ 2, 4 ]
+ - [ 4, 2 ]
+ - [ 4, 4 ]
+ - [ 4, 8 ]
+ - [ 8, 4 ]
+ - [ 8, 8 ]
+ - WorkGroup:
+ - [ 8, 32, 1 ]
+ - [ 8, 8, 1 ]
+ - [ 8, 16, 1 ]
+ - [ 16, 8, 1 ]
+ - [ 16, 16, 1 ]
+ - [ 32, 8, 1 ]
+ - [ 32, 16, 1 ]
+ - WorkGroupMapping: [1, 8]
+ - DepthU: [ 8, 16 ]
+ - VectorWidth: [1,2,4]
+ - GlobalReadVectorWidth: [4]
+ - LdsPadB: [0,1,2,4]
+ BenchmarkForkParameters:
+ JoinParameters:
+ BenchmarkJoinParameters:
+ BenchmarkFinalParameters:
+ - ProblemSizes:
+ - Exact: [ 196, 256, 64, 1024 ]
+ - Exact: [ 784, 512, 64, 128 ]
+ - Exact: [ 49, 512, 64, 2048 ]
+ - Exact: [ 196, 1024, 64, 256 ]
+ - Exact: [ 49, 2048, 64, 512 ]
+ - Exact: [ 3025, 256, 64, 64 ]
+ - Exact: [ 3025, 64, 64, 64 ]
+
########################################
########################################
