Skip to content

Commit 0d9b478

Browse files
committed
[mlir] Reduce the number of iterations in async microbenchmarks
Differential Revision: https://reviews.llvm.org/D112609
1 parent 1a2a7cc commit 0d9b478

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

mlir/test/Integration/Dialect/Async/CPU/microbench-linalg-async-parallel-for.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -57,7 +57,7 @@ func @entry() {
5757
%f4 = arith.constant 4.0 : f32
5858
%c0 = arith.constant 0 : index
5959
%c1 = arith.constant 1 : index
60-
%cM = arith.constant 1000 : index
60+
%cN = arith.constant 50 : index
6161

6262
//
6363
// Sanity check for the function under test.
@@ -109,7 +109,7 @@ func @entry() {
109109
//
110110

111111
%t0 = call @rtclock() : () -> f64
112-
scf.for %i = %c0 to %cM step %c1 {
112+
scf.for %i = %c0 to %cN step %c1 {
113113
call @linalg_generic(%LHS0, %RHS0, %DST0)
114114
: (memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>) -> ()
115115
}

mlir/test/Integration/Dialect/Async/CPU/microbench-scf-async-parallel-for.mlir

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,7 +78,7 @@ func @entry() {
7878
%f4 = arith.constant 4.0 : f32
7979
%c0 = arith.constant 0 : index
8080
%c1 = arith.constant 1 : index
81-
%cM = arith.constant 1000 : index
81+
%cN = arith.constant 50 : index
8282

8383
//
8484
// Sanity check for the function under test.
@@ -130,7 +130,7 @@ func @entry() {
130130
//
131131

132132
%t0 = call @rtclock() : () -> f64
133-
scf.for %i = %c0 to %cM step %c1 {
133+
scf.for %i = %c0 to %cN step %c1 {
134134
call @scf_parallel(%LHS0, %RHS0, %DST0)
135135
: (memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>) -> ()
136136
}

0 commit comments

Comments
 (0)