[compiler] update byteirRegisterDialectExtensions

bytedance · May 6, 2024 · 82fb2ec · 82fb2ec
1 parent 55ae0f2
commit 82fb2ec
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 8 deletions.
diff --git a/compiler/lib/CAPI/Dialects.cpp b/compiler/lib/CAPI/Dialects.cpp
@@ -23,6 +23,7 @@
 #include "byteir/Dialect/Cat/IR/CatDialect.h"
 #include "byteir/Dialect/Ccl/IR/CclOps.h"
 #include "byteir/Dialect/Ccl/TransformOps/CclTransformOps.h"
+#include "byteir/Dialect/GPU/TransformOps/GPUExtTransformOps.h"
 #include "byteir/Dialect/Linalg/TransformOps/LinalgExtTransformOps.h"
 #include "byteir/Dialect/Tensor/IR/TilingInterfaceImpl.h"
 #include "byteir/Dialect/Transform/IR/TransformExtOps.h"
@@ -45,5 +46,6 @@ void byteirRegisterDialectExtensions(MlirContext context) {
   linalg_ext::registerTransformDialectExtension(registry);
   transform_ext::registerTransformDialectExtension(registry);
   tensor_ext::registerTilingInterfaceExternalModels(registry);
+  gpu_ext::registerTransformDialectExtension(registry);
   unwrap(context)->appendDialectRegistry(registry);
 }
diff --git a/compiler/lib/Dialect/GPU/TransformOps/CMakeLists.txt b/compiler/lib/Dialect/GPU/TransformOps/CMakeLists.txt
@@ -12,17 +12,10 @@ add_mlir_dialect_library(MLIRGPUExtTransformOps
   LINK_LIBS PUBLIC
   MLIRGPUDialect
   MLIRGPUTransforms
-  MLIRGPUTransformOps
   MLIRIR
   MLIRParser
   MLIRSideEffectInterfaces
   MLIRTransformDialect
-  MLIRVectorDialect
-  MLIRVectorTransforms
   MLIRPDLDialect
   MLIRSCFDialect
-
-  # ConversionPatterns
-  MLIRNVGPUToNVVM
-  MLIRGPUToNVVMTransforms
-  )  
+)