Skeleton for the pipeline pass

lib/Dialect/TritonGPU/Transforms/CMakeLists.txt

@@ -1,10 +1,12 @@
-add_mlir_dialect_library(TritonGPUConversion
+add_mlir_dialect_library(TritonGPUTransforms
+  Pipeline.cpp
   TritonGPUConversion.cpp
 
-  # ADDITIONAL_HEADER_DIRS
+  DEPENDS
+  TritonGPUTransformsIncGen
 
   LINK_LIBS PUBLIC
   TritonIR
   TritonGPUIR
-  # MLIRTransformUtils
+  MLIRTransformUtils
 )

lib/Dialect/TritonGPU/Transforms/Pipeline.cpp

@@ -0,0 +1,21 @@
+#include "triton/Dialect/TritonGPU/IR/Dialect.h"
+#include "triton/Dialect/TritonGPU/Transforms/Passes.h"
+
+using namespace mlir;
+
+#define GEN_PASS_CLASSES
+#include "triton/Dialect/TritonGPU/Transforms/Passes.h.inc"
+
+namespace {
+struct PipelinePass : public TritonGPUPipelineBase<PipelinePass> {
+  void runOnOperation() override {
+    getOperation()->walk([&](scf::ForOp forOp) {
+
+    });
+  }
+};
+} // anonymous namespace
+
+std::unique_ptr<Pass> mlir::createTritonGPUPipelinePass() {
+  return std::make_unique<PipelinePass>();
+}