aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorlewuathe <lewuathe@me.com>2022-06-25 08:59:13 +0900
committerlewuathe <lewuathe@me.com>2022-06-25 09:30:43 +0900
commitbd861056a528c71ea45cabcfe7b89647732d9afc (patch)
treec676cfbd6a0545f7ed8c24520de459d6e213750e
parent96f6ec5090c2f7a1e4804693cbb84c29c574b3de (diff)
downloadllvm-bd861056a528c71ea45cabcfe7b89647732d9afc.zip
llvm-bd861056a528c71ea45cabcfe7b89647732d9afc.tar.gz
llvm-bd861056a528c71ea45cabcfe7b89647732d9afc.tar.bz2
[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
-rw-r--r--mlir/test/Dialect/SCF/loop-unroll.mlir2
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/test/Dialect/SCF/loop-unroll.mlir b/mlir/test/Dialect/SCF/loop-unroll.mlir
index dc2c07f..7c029cb 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