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

@@ -42,4 +42,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

@@ -42,7 +42,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

mlir/test/Dialect/XeGPU/XeGPUOps.mlir

@@ -1,3 +1,4 @@
+// REQUIRES: xegpu-dialect-enabled
 // RUN: mlir-opt %s | FileCheck %s
 // Verify the printed output can be parsed.
 // RUN: mlir-opt %s | mlir-opt | FileCheck %s

mlir/test/Dialect/XeGPU/invalid.mlir

@@ -1,3 +1,4 @@
+// REQUIRES: xegpu-dialect-enabled
 // RUN: mlir-opt %s -split-input-file -verify-diagnostics
 
 // -----

mlir/test/Dialect/XeGPU/xegpu-fold-alias-ops.mlir

@@ -1,3 +1,4 @@
+// REQUIRES: xegpu-dialect-enabled
 // RUN: mlir-opt -xegpu-fold-alias-ops -split-input-file %s | FileCheck %s
 
 func.func @fold_subview_with_xegpu_create_nd_tdesc(%arg0 : memref<256x256xf32>, %arg1 : index, %arg2 : index, %arg3 : index, %arg4 : index) ->(!xegpu.tensor_desc<8x16xf32>) {

mlir/test/lit.site.cfg.py.in

@@ -69,6 +69,10 @@ config.riscv_vector_emulator_options = "@RISCV_VECTOR_EMULATOR_OPTIONS@"
 config.riscv_emulator_lli_executable = "@RISCV_EMULATOR_LLI_EXECUTABLE@"
 config.riscv_emulator_utils_lib_dir = "@RISCV_EMULATOR_UTILS_LIB_DIR@"
 
+# Add feature to enable/disable XeGPU Dialect tests
+if "@MLIR_DIALECT_XEGPU_ENABLE@" == "ON":
+    config.available_features.add("xegpu-dialect-enabled")
+
 import lit.llvm
 lit.llvm.initialize(lit_config, config)