Skeleton for the pipeline pass

include/triton/Dialect/TritonGPU/CMakeLists.txt

@@ -1 +1,2 @@
 add_subdirectory(IR)
+add_subdirectory(Transforms)

include/triton/Dialect/TritonGPU/Transforms/CMakeLists.txt

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

include/triton/Dialect/TritonGPU/Transforms/Passes.h

@@ -0,0 +1,14 @@
+#ifndef TRITON_DIALECT_TRITONGPU_TRANSFORMS_PASSES_H_
+#define TRITON_DIALECT_TRITONGPU_TRANSFORMS_PASSES_H_
+
+#include "mlir/Pass/Pass.h"
+
+namespace mlir {
+std::unique_ptr<Pass> createTritonGPUPipelinePass();
+
+// /// Generate the code for registering passes.
+// #define GEN_PASS_REGISTRATION
+// #include "triton/Dialect/TritonGPU/Transforms/Passes.h.inc"
+
+} // namespace mlir
+#endif

include/triton/Dialect/TritonGPU/Transforms/Passes.td

@@ -0,0 +1,27 @@
+#ifndef TRITONGPU_PASSES
+#define TRITONGPU_PASSES
+
+include "mlir/Pass/PassBase.td"
+
+def TritonGPUPipeline : Pass<"tritongpu-pipeline", "mlir::FuncOp"> {
+  let summary = "pipeline";
+
+  let description = [{
+    scf.for() {
+      %a = load %a_ptr;
+      %b = load %b_ptr;
+
+      %d = dot %a, %b, %c;
+    }
+
+    =>
+
+    ...
+  }];
+
+  let constructor = "mlir::triton::gpu::createPipelinePass";
+
+  let dependentDialects = ["mlir::triton::gpu::TritonGPUDialect"];
+}
+
+#endif