diff options
author | lewuathe <lewuathe@me.com> | 2022-06-25 08:59:13 +0900 |
---|---|---|
committer | lewuathe <lewuathe@me.com> | 2022-06-25 09:30:43 +0900 |
commit | bd861056a528c71ea45cabcfe7b89647732d9afc (patch) | |
tree | c676cfbd6a0545f7ed8c24520de459d6e213750e | |
parent | 96f6ec5090c2f7a1e4804693cbb84c29c574b3de (diff) | |
download | llvm-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.mlir | 2 |
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 |