[Mlir-commits] [mlir] bd86105 - [mlir][affine] Rigorous check for loop unrolling store operation

[llvmlistbot at llvm.org]

Author: lewuathe
Date: 2022-06-25T09:30:43+09:00
New Revision: bd861056a528c71ea45cabcfe7b89647732d9afc

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

LOG: [mlir][affine] Rigorous check for loop unrolling store operation

Static loop unrolling does not change the operation type. We can rigorously make sure to use affine.store in the check.

Reviewed By: ftynse

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

Added: 
    

Modified: 
    mlir/test/Dialect/SCF/loop-unroll.mlir

Removed: 
    


################################################################################
diff  --git a/mlir/test/Dialect/SCF/loop-unroll.mlir b/mlir/test/Dialect/SCF/loop-unroll.mlir
index dc2c07f291e76..7c029cb29df74 100644
--- a/mlir/test/Dialect/SCF/loop-unroll.mlir
+++ b/mlir/test/Dialect/SCF/loop-unroll.mlir
@@ -271,7 +271,7 @@ func.func @static_loop_unroll_up_to_factor(%arg0 : memref<?xf32>) {
 //   UNROLL-UP-TO-DAG:  %[[C0:.*]] = arith.constant 0 : index
 //   UNROLL-UP-TO-DAG:  %[[C2:.*]] = arith.constant 2 : index
 //   UNROLL-UP-TO-NEXT: %[[V0:.*]] = affine.apply {{.*}}
-//   UNROLL-UP-TO-NEXT: store %{{.*}}, %[[MEM]][%[[V0]]] : memref<?xf32>
+//   UNROLL-UP-TO-NEXT: affine.store %{{.*}}, %[[MEM]][%[[V0]]] : memref<?xf32>
 //   UNROLL-UP-TO-NEXT: %[[V1:.*]] = affine.apply {{.*}}
 //   UNROLL-UP-TO-NEXT: affine.store %{{.*}}, %[[MEM]][%[[V1]]] : memref<?xf32>
 //   UNROLL-UP-TO-NEXT: return