[AnnotateScop] new pass annotates scop.ignored

Author: kumasento

tools/polymer/include/polymer/CMakeLists.txt

@@ -0,0 +1 @@
+add_subdirectory(Transforms)

tools/polymer/include/polymer/Transforms/CMakeLists.txt

@@ -0,0 +1,3 @@
+set(LLVM_TARGET_DEFINITIONS Passes.td)
+mlir_tablegen(Passes.h.inc -gen-pass-decls -name Transforms)
+add_public_tablegen_target(PolymerTransformsIncGen)

tools/polymer/include/polymer/Transforms/Passes.h

@@ -0,0 +1,17 @@
+//===- Passes.h - Include Tblgen pass defs ------------C++-===//
+#ifndef POLYMER_TRANSFORMS_PASSES_H
+#define POLYMER_TRANSFORMS_PASSES_H
+
+#include "mlir/Pass/Pass.h"
+
+namespace polymer {
+
+std::unique_ptr<mlir::OperationPass<mlir::FuncOp>> createAnnotateScopPass();
+
+/// Generate the code for registering passes.
+#define GEN_PASS_REGISTRATION
+#include "polymer/Transforms/Passes.h.inc"
+
+} // namespace polymer
+
+#endif

tools/polymer/include/polymer/Transforms/Passes.td

@@ -0,0 +1,21 @@
+#ifndef POLYMER_TRANSFORMS_PASSES
+#define POLYMER_TRANSFORMS_PASSES
+
+include "mlir/Pass/PassBase.td"
+include "mlir/Rewrite/PassUtil.td"
+
+def AnnotateScop : FunctionPass<"annotate-scop"> {
+  let summary = "Annotate scop.ignore to unselected functions.";
+  let constructor = "polymer::createAnnotateScopPass()";
+
+  let options = [
+    ListOption<
+      "includedFunctions",
+      "functions",
+      "std::string",   
+      "A list of functions to be included.",
+      "llvm::cl::ZeroOrMore, llvm::cl::MiscFlags::CommaSeparated">
+  ];
+}
+
+#endif

tools/polymer/lib/Transforms/AnnotateScop.cc

@@ -0,0 +1,34 @@
+//===- AnnotateScop.cc --------------------------------------*- C++ -*-===//
+
+#include "PassDetail.h"
+
+#include "mlir/Dialect/Affine/IR/AffineOps.h"
+#include "mlir/Dialect/MemRef/IR/MemRef.h"
+#include "mlir/Dialect/SCF/SCF.h"
+#include "mlir/Transforms/LoopUtils.h"
+#include "mlir/Transforms/Passes.h"
+#include "mlir/Transforms/RegionUtils.h"
+#include "llvm/Support/Debug.h"
+
+using namespace mlir;
+using namespace llvm;
+using namespace polymer;
+
+namespace {
+struct AnnotateScop : public polymer::AnnotateScopBase<AnnotateScop> {
+  void runOnFunction() override {
+    FuncOp f = getOperation();
+    OpBuilder b(f.getContext());
+
+    for (auto &name : includedFunctions)
+      if (name == f.getName())
+        return;
+
+    f->setAttr("scop.ignored", b.getUnitAttr());
+  } // namespace
+};
+} // namespace
+
+std::unique_ptr<OperationPass<FuncOp>> polymer::createAnnotateScopPass() {
+  return std::make_unique<AnnotateScop>();
+}

tools/polymer/lib/Transforms/CMakeLists.txt

@@ -6,12 +6,13 @@ add_mlir_conversion_library(PolymerTransforms
   LoopAnnotate.cc
   LoopExtract.cc
   FoldSCFIf.cc
+  AnnotateScop.cc
 
   ADDITIONAL_HEADER_DIRS
   "${POLYMER_MAIN_INCLUDE_DIR}/polymer/Transforms"
 
   DEPENDS
-  ${POLYMER_TRANSFORM_DEPENDS}
+  PolymerTransformsIncGen
 
   LINK_LIBS PUBLIC
   MLIRAffine

tools/polymer/lib/Transforms/PassDetail.h

@@ -0,0 +1,14 @@
+//===- PassDetail.h - Transforms Pass class details -------------*- C++ -*-===//
+
+#ifndef POLYMER_TRANSFORMS_PASSDETAIL_H_
+#define POLYMER_TRANSFORMS_PASSDETAIL_H_
+
+#include "mlir/Pass/Pass.h"
+#include "polymer/Transforms/Passes.h"
+
+namespace polymer {
+#define GEN_PASS_CLASSES
+#include "polymer/Transforms/Passes.h.inc"
+} // namespace polymer
+
+#endif

tools/polymer/test/polymer-opt/AnnotateScop/annotate-single.mlir

@@ -0,0 +1,12 @@
+// RUN: polymer-opt %s -annotate-scop="functions=foo" | FileCheck %s
+
+func @foo() {
+  return
+} 
+ 
+func @bar() {
+  return
+} 
+
+// CHECK: func @foo() {
+// CHECK: func @bar() attributes {scop.ignored} {

tools/polymer/tools/polymer-opt/polymer-opt.cc

@@ -9,6 +9,7 @@
 #include "polymer/Transforms/FoldSCFIf.h"
 #include "polymer/Transforms/LoopAnnotate.h"
 #include "polymer/Transforms/LoopExtract.h"
+#include "polymer/Transforms/Passes.h"
 #include "polymer/Transforms/PlutoTransform.h"
 #include "polymer/Transforms/Reg2Mem.h"
 #include "polymer/Transforms/ScopStmtOpt.h"
@@ -61,6 +62,7 @@ int main(int argc, char *argv[]) {
   registerLoopAnnotatePasses();
   registerLoopExtractPasses();
   registerFoldSCFIfPass();
+  registerAnnotateScopPass();
 
   // Register any pass manager command line options.
   registerMLIRContextCLOptions();