EISW-163563 Add CMake option to conditionally enable XeGPU dialect

Introduced a CMake option MLIR_DIALECT_XEGPU_ENABLE (default OFF) to optionally include the XeGPU dialect in the build.

Signed-off-by: Surendar Rama Sitaraman <[email protected]>

Author: sramasit

mlir/include/mlir/Dialect/CMakeLists.txt

@@ -44,4 +44,6 @@ add_subdirectory(UB)
 add_subdirectory(Utils)
 add_subdirectory(Vector)
 add_subdirectory(X86Vector)
-add_subdirectory(XeGPU)
+if(MLIR_DIALECT_XEGPU_ENABLE)
+  add_subdirectory(XeGPU)
+endif()

mlir/include/mlir/InitAllDialects.h

@@ -94,7 +94,9 @@
 #include "mlir/Dialect/Vector/Transforms/BufferizableOpInterfaceImpl.h"
 #include "mlir/Dialect/Vector/Transforms/SubsetOpInterfaceImpl.h"
 #include "mlir/Dialect/X86Vector/X86VectorDialect.h"
+#ifdef MLIR_DIALECT_XEGPU_ENABLE
 #include "mlir/Dialect/XeGPU/IR/XeGPU.h"
+#endif
 #include "mlir/IR/Dialect.h"
 #include "mlir/Interfaces/CastInterfaces.h"
 #include "mlir/Target/LLVM/NVVM/Target.h"
@@ -149,10 +151,13 @@ inline void registerAllDialects(DialectRegistry &registry) {
                   transform::TransformDialect,
                   ub::UBDialect,
                   vector::VectorDialect,
-                  x86vector::X86VectorDialect,
-                  xegpu::XeGPUDialect>();
+                  x86vector::X86VectorDialect>();
   // clang-format on
 
+#ifdef MLIR_DIALECT_XEGPU_ENABLE
+  register.insert<xegpu::XeGPUDialect>();
+#endif
+
   // Register all external models.
   affine::registerValueBoundsOpInterfaceExternalModels(registry);
   arith::registerBufferDeallocationOpInterfaceExternalModels(registry);

mlir/include/mlir/InitAllPasses.h

@@ -45,7 +45,9 @@
 #include "mlir/Dialect/Tosa/Transforms/Passes.h"
 #include "mlir/Dialect/Transform/Transforms/Passes.h"
 #include "mlir/Dialect/Vector/Transforms/Passes.h"
+#ifdef MLIR_DIALECT_XEGPU_ENABLE
 #include "mlir/Dialect/XeGPU/Transforms/Passes.h"
+#endif
 #include "mlir/Transforms/Passes.h"
 
 #include <cstdlib>
@@ -94,7 +96,10 @@ inline void registerAllPasses() {
   arm_sme::registerArmSMEPasses();
   arm_sve::registerArmSVEPasses();
   emitc::registerEmitCPasses();
+
+#ifdef MLIR_DIALECT_XEGPU_ENABLE
   xegpu::registerXeGPUPasses();
+#endif
 
   // Dialect pipelines
   bufferization::registerBufferizationPipelines();

mlir/lib/Dialect/CMakeLists.txt

@@ -44,7 +44,9 @@ add_subdirectory(UB)
 add_subdirectory(Utils)
 add_subdirectory(Vector)
 add_subdirectory(X86Vector)
-add_subdirectory(XeGPU)
+if(MLIR_DIALECT_XEGPU_ENABLE)
+  add_subdirectory(XeGPU)
+endif()
 
 set(LLVM_OPTIONAL_SOURCES
   Traits.cpp