Skip to content

Commit d3584c7

Browse files
kumasentoivanradanov
authored andcommitted
[ExtractScopStmt/PlutoOpt] supports scop.ignored
1 parent 372aa45 commit d3584c7

File tree

2 files changed

+6
-2
lines changed

2 files changed

+6
-2
lines changed

tools/polymer/lib/Transforms/ExtractScopStmt.cc

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -511,7 +511,11 @@ class ExtractScopStmtPass
511511
OpBuilder b(m.getContext());
512512

513513
SmallVector<mlir::FuncOp, 4> funcs;
514-
m.walk([&](mlir::FuncOp f) { funcs.push_back(f); });
514+
m.walk([&](mlir::FuncOp f) {
515+
if (f->hasAttr("scop.ignored"))
516+
return;
517+
funcs.push_back(f);
518+
});
515519

516520
unsigned numCallees = 0;
517521
for (mlir::FuncOp f : funcs) {

tools/polymer/lib/Transforms/PlutoTransform.cc

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ class PlutoTransformPass
156156
llvm::DenseMap<mlir::FuncOp, mlir::FuncOp> funcMap;
157157

158158
m.walk([&](mlir::FuncOp f) {
159-
if (!f->getAttr("scop.stmt"))
159+
if (!f->getAttr("scop.stmt") && !f->hasAttr("scop.ignored"))
160160
funcOps.push_back(f);
161161
});
162162

0 commit comments

Comments
 (0)