more TritonGPU unit tests

2022-06-05 14:25:09 +08:00
parent 55cf9a0a97
commit d5eca56cf3
3 changed files with 34 additions and 5 deletions
--- a/bin/triton-opt.cpp
+++ b/bin/triton-opt.cpp
@@ -11,7 +11,10 @@ int main(int argc, char **argv) {

  mlir::DialectRegistry registry;
  registry.insert<mlir::triton::TritonDialect,
-                  mlir::triton::gpu::TritonGPUDialect>();
+                  mlir::triton::gpu::TritonGPUDialect,
+                  mlir::arith::ArithmeticDialect,
+                  mlir::StandardOpsDialect,
+                  mlir::scf::SCFDialect>();

  return mlir::asMainReturnCode(
      mlir::MlirOptMain(argc, argv, "Triton (GPU) optimizer driver\n", registry)