[mlir] Reduce the number of iterations in async microbenchmarks

Differential Revision: https://reviews.llvm.org/D112609

diff --git a/mlir/test/Integration/Dialect/Async/CPU/microbench-linalg-async-parallel-for.mlir b/mlir/test/Integration/Dialect/Async/CPU/microbench-linalg-async-parallel-for.mlir
index 1b64931..7528319 100644 (file)
--- a/mlir/test/Integration/Dialect/Async/CPU/microbench-linalg-async-parallel-for.mlir
+++ b/mlir/test/Integration/Dialect/Async/CPU/microbench-linalg-async-parallel-for.mlir
@@ -57,7 +57,7 @@ func @entry() {
   %f4 = arith.constant 4.0 : f32
   %c0 = arith.constant 0 : index
   %c1 = arith.constant 1 : index
-  %cM = arith.constant 1000 : index
+  %cN = arith.constant 50 : index
 
   //
   // Sanity check for the function under test.
@@ -109,7 +109,7 @@ func @entry() {
   //
 
   %t0 = call @rtclock() : () -> f64
-  scf.for %i = %c0 to %cM step %c1 {
+  scf.for %i = %c0 to %cN step %c1 {
     call @linalg_generic(%LHS0, %RHS0, %DST0)
       : (memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>) -> ()
   }

diff --git a/mlir/test/Integration/Dialect/Async/CPU/microbench-scf-async-parallel-for.mlir b/mlir/test/Integration/Dialect/Async/CPU/microbench-scf-async-parallel-for.mlir
index 8eca332..4ca1bc9 100644 (file)
--- a/mlir/test/Integration/Dialect/Async/CPU/microbench-scf-async-parallel-for.mlir
+++ b/mlir/test/Integration/Dialect/Async/CPU/microbench-scf-async-parallel-for.mlir
@@ -78,7 +78,7 @@ func @entry() {
   %f4 = arith.constant 4.0 : f32
   %c0 = arith.constant 0 : index
   %c1 = arith.constant 1 : index
-  %cM = arith.constant 1000 : index
+  %cN = arith.constant 50 : index
 
   //
   // Sanity check for the function under test.
@@ -130,7 +130,7 @@ func @entry() {
   //
 
   %t0 = call @rtclock() : () -> f64
-  scf.for %i = %c0 to %cM step %c1 {
+  scf.for %i = %c0 to %cN step %c1 {
     call @scf_parallel(%LHS0, %RHS0, %DST0)
       : (memref<?x?xf32>, memref<?x?xf32>, memref<?x?xf32>) -> ()
   }