From 0d9b478932810176e844a97a5ab5f79975730bb7 Mon Sep 17 00:00:00 2001
From: Eugene Zhulenev <ezhulenev@google.com>
Date: Wed, 27 Oct 2021 03:18:47 -0700
Subject: [PATCH] [mlir] Reduce the number of iterations in async
 microbenchmarks

Differential Revision: https://reviews.llvm.org/D112609
---
 .../Dialect/Async/CPU/microbench-linalg-async-parallel-for.mlir       | 4 ++--
 .../Dialect/Async/CPU/microbench-scf-async-parallel-for.mlir          | 4 ++--
 2 files changed, 4 insertions(+), 4 deletions(-)

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
--- 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
--- 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>) -> ()
   }
-- 
2.7.4