TritonGPU verifier

This commit is contained in:
Yan Da
2022-05-24 19:48:56 +08:00
parent 36c45ec687
commit a2c9f919a8
10 changed files with 131 additions and 4 deletions

View File

@@ -8,6 +8,8 @@ def TritonGPU_Dialect : Dialect {
let cppNamespace = "::mlir::triton::gpu";
let hasOperationAttrVerify = 1;
let description = [{
Triton GPU Dialect.
}];

View File

@@ -9,6 +9,8 @@ std::unique_ptr<Pass> createTritonGPUPipelinePass(int numStages);
namespace triton {
namespace gpu {
std::unique_ptr<Pass> createCombineOpsPass();
std::unique_ptr<Pass> createTritonGPUVerifier();
}
}

View File

@@ -51,4 +51,14 @@ def TritonGPUCombineOps : Pass<"tritongpu-combine", "mlir::ModuleOp"> {
"mlir::triton::TritonDialect"];
}
def TritonGPUVerifier : Pass<"tritongpu-verifier", "mlir::ModuleOp"> {
let summary = "verify TritonGPU IR";
let description = [{}];
let constructor = "mlir::triton::gpu::createTritonGPUVerifier";
let dependentDialects = ["mlir::triton::gpu::TritonGPUDialect"];
}
#endif