[Reg2Mem] understands scop.ignored

kumasento · Oct 21, 2021 · aa454e9 · aa454e9
1 parent a83a0c8
commit aa454e9
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 0 deletions.
diff --git a/lib/Transforms/Reg2Mem.cc b/lib/Transforms/Reg2Mem.cc
@@ -315,6 +315,9 @@ class RegToMemPass
     mlir::FuncOp f = getOperation();
     auto builder = OpBuilder(f.getContext());
 
+    if (f->hasAttr("scop.ignored"))
+      return;
+
     separateAffineIfBlocks(f, builder);
     demoteRegisterToMemory(f, builder);
   }

diff --git a/test/polymer-opt/Reg2Mem/ignored.mlir b/test/polymer-opt/Reg2Mem/ignored.mlir
@@ -0,0 +1,14 @@
+// RUN: polymer-opt %s -reg2mem | FileCheck %s
+func @foo(%A: memref<?xf32>) attributes {scop.ignored} {
+  %0 = affine.load %A[0] : memref<?xf32>
+  affine.for %i = 1 to 30 {
+    affine.store %0, %A[%i] : memref<?xf32>
+  }
+
+  return
+}
+
+// CHECK: func @foo
+// CHECK-NEXT:   %[[v0:.*]] = affine.load 
+// CHECK-NEXT:   affine.for %{{.*}} = 1 to 30 
+// CHECK-NEXT:     affine.store %[[v0]]