TritonGPU combiner

2022-05-16 19:16:01 +08:00
parent 0e68e6eb59
commit e3916c3a46
9 changed files with 109 additions and 118 deletions
--- a/lib/Dialect/TritonGPU/Transforms/CMakeLists.txt
+++ b/lib/Dialect/TritonGPU/Transforms/CMakeLists.txt
@@ -1,9 +1,15 @@
+set(LLVM_TARGET_DEFINITIONS Combine.td)
+mlir_tablegen(TritonGPUCombine.inc -gen-rewriters)
+add_public_tablegen_target(TritonGPUCombineIncGen)
+
 add_mlir_dialect_library(TritonGPUTransforms
+  Combine.cpp
  Pipeline.cpp
  TritonGPUConversion.cpp

  DEPENDS
  TritonGPUTransformsIncGen
+  TritonGPUCombineIncGen

  LINK_LIBS PUBLIC
  TritonIR