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.

Author: sramasit

mlir/CMakeLists.txt

@@ -191,6 +191,15 @@ set(MLIR_PDLL_TABLEGEN_TARGET "${MLIR_PDLL_TABLEGEN_TARGET}" CACHE INTERNAL "")
 set(MLIR_SRC_SHARDER_TABLEGEN_EXE "${MLIR_SRC_SHARDER_TABLEGEN_EXE}" CACHE INTERNAL "")
 set(MLIR_SRC_SHARDER_TABLEGEN_TARGET "${MLIR_SRC_SHARDER_TABLEGEN_TARGET}" CACHE INTERNAL "")
 
+# XeGPU Dialect Option (Default OFF)
+option(MLIR_DIALECT_XEGPU_ENABLE
+       "Enable the XeGPU dialect."
+       OFF)
+
+if(MLIR_DIALECT_XEGPU_ENABLE)
+  add_compile_definitions(MLIR_DIALECT_XEGPU_ENABLE)
+endif()
+
 add_subdirectory(include/mlir)
 add_subdirectory(lib)
 # C API needs all dialects for registration, but should be built before tests.

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

@@ -92,7 +92,9 @@
 #include "mlir/Dialect/Vector/Transforms/BufferizableOpInterfaceImpl.h"
 #include "mlir/Dialect/Vector/Transforms/SubsetOpInterfaceImpl.h"
 #include "mlir/Dialect/X86Vector/X86VectorDialect.h"
-/* #include "mlir/Dialect/XeGPU/IR/XeGPU.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"
@@ -147,10 +149,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

@@ -44,7 +44,9 @@
 #include "mlir/Dialect/Tosa/Transforms/Passes.h"
 #include "mlir/Dialect/Transform/Transforms/Passes.h"
 #include "mlir/Dialect/Vector/Transforms/Passes.h"
-/* #include "mlir/Dialect/XeGPU/Transforms/Passes.h" */
+#ifdef MLIR_DIALECT_XEGPU_ENABLE 
+#include "mlir/Dialect/XeGPU/Transforms/Passes.h"
+#endif
 #include "mlir/Transforms/Passes.h"
 
 #include <cstdlib>
@@ -92,7 +94,10 @@ inline void registerAllPasses() {
   arm_sme::registerArmSMEPasses();
   arm_sve::registerArmSVEPasses();
   emitc::registerEmitCPasses();
-  /* xegpu::registerXeGPUPasses(); */
+    
+#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