[Mlir-commits] [mlir] 0d9b478 - [mlir] Reduce the number of iterations in async microbenchmarks

[Eugene Zhulenev]

Author: Eugene Zhulenev
Date: 2021-10-27T03:20:06-07:00
New Revision: 0d9b478932810176e844a97a5ab5f79975730bb7

URL: https://github.com/llvm/llvm-project/commit/0d9b478932810176e844a97a5ab5f79975730bb7
DIFF: https://github.com/llvm/llvm-project/commit/0d9b478932810176e844a97a5ab5f79975730bb7.diff

LOG: [mlir] Reduce the number of iterations in async microbenchmarks

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

Added: 
    

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

Removed: 
    


################################################################################
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 1b64931f7806a..7528319c52acd 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 8eca332fc8ece..4ca1bc9029aaa 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>) -> ()
   }