[Reg2Mem] understands scop.ignored

Author: kumasento

tools/polymer/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);
   }

tools/polymer/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]]