[ExtractScopStmt/PlutoOpt] supports scop.ignored

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

Hide file tree

Showing 2 changed files with 6 additions and 2 deletions.
diff --git a/lib/Transforms/ExtractScopStmt.cc b/lib/Transforms/ExtractScopStmt.cc
@@ -510,7 +510,11 @@ class ExtractScopStmtPass
  OpBuilder b(m.getContext());
 
  SmallVector<mlir::FuncOp, 4> funcs;
- m.walk([&](mlir::FuncOp f) { funcs.push_back(f); });
+ m.walk([&](mlir::FuncOp f) {
+ if (f->hasAttr("scop.ignored"))
+ return;
+ funcs.push_back(f);
+ });
 
  unsigned numCallees = 0;
  for (mlir::FuncOp f : funcs) {

diff --git a/lib/Transforms/PlutoTransform.cc b/lib/Transforms/PlutoTransform.cc
@@ -156,7 +156,7 @@ class PlutoTransformPass
  llvm::DenseMap<mlir::FuncOp, mlir::FuncOp> funcMap;
 
  m.walk([&](mlir::FuncOp f) {
- if (!f->getAttr("scop.stmt"))
+ if (!f->getAttr("scop.stmt") && !f->hasAttr("scop.ignored"))
  funcOps.push_back(f);
  });