Skeleton for the pipeline pass

This commit is contained in:
Yan Da
2022-05-11 16:13:53 +08:00
parent 96876a46d1
commit 1a4fbed25b
9 changed files with 344 additions and 12 deletions

View File

@@ -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>();
}