EISW-170058 Add CMake option to conditionally enable VectorToXeGPU conversion

Author: sramasit

mlir/include/mlir/Conversion/CMakeLists.txt

@@ -10,4 +10,7 @@ add_mlir_doc(Passes ConversionPasses ./ -gen-pass-doc)
 if(MLIR_CONVERSION_TOSATOTENSOR_ENABLE)
   add_subdirectory(TosaToTensor)
 endif()
+if(MLIR_DIALECT_XEGPU_ENABLE)
+  add_subdirectory(VectorToXeGPU)
+endif()
 add_subdirectory(ConvertToLLVM)

mlir/include/mlir/Conversion/Passes.h

@@ -81,7 +81,9 @@
 #include "mlir/Conversion/VectorToLLVM/ConvertVectorToLLVMPass.h"
 #include "mlir/Conversion/VectorToSCF/VectorToSCF.h"
 #include "mlir/Conversion/VectorToSPIRV/VectorToSPIRVPass.h"
+#ifdef MLIR_DIALECT_XEGPU_ENABLE
 #include "mlir/Conversion/VectorToXeGPU/VectorToXeGPU.h"
+#endif
 
 namespace mlir {
 

mlir/include/mlir/Conversion/Passes.td

@@ -1465,6 +1465,7 @@ def ConvertVectorToSPIRV : Pass<"convert-vector-to-spirv"> {
 // VectorToXeGPU
 //===----------------------------------------------------------------------===//
 
+#ifdef MLIR_DIALECT_XEGPU_ENABLE
 def ConvertVectorToXeGPU : Pass<"convert-vector-to-xegpu"> {
   let summary = "Lower the operations from the vector dialect into the XeGPU "
                 "dialect";
@@ -1474,5 +1475,6 @@ def ConvertVectorToXeGPU : Pass<"convert-vector-to-xegpu"> {
     "vector::VectorDialect", "xegpu::XeGPUDialect"
   ];
 }
+#endif
 
 #endif // MLIR_CONVERSION_PASSES

mlir/lib/Conversion/CMakeLists.txt

@@ -92,4 +92,6 @@ add_subdirectory(VectorToSCF)
 if(MLIR_DIALECT_SPIRV_ENABLE)
   add_subdirectory(VectorToSPIRV)
 endif()
-add_subdirectory(VectorToXeGPU)
+if(MLIR_DIALECT_XEGPU_ENABLE)
+  add_subdirectory(VectorToXeGPU)
+endif()

mlir/lib/Conversion/VectorToXeGPU/CMakeLists.txt

@@ -1,16 +1,18 @@
-add_mlir_conversion_library(MLIRVectorToXeGPU
-  VectorToXeGPU.cpp
+if(MLIR_DIALECT_XEGPU_ENABLE)
+  add_mlir_conversion_library(MLIRVectorToXeGPU
+    VectorToXeGPU.cpp
 
-  ADDITIONAL_HEADER_DIRS
-  ${MLIR_MAIN_INCLUDE_DIR}/mlir/Conversion/VectorToXeGPU
+    ADDITIONAL_HEADER_DIRS
+    ${MLIR_MAIN_INCLUDE_DIR}/mlir/Conversion/VectorToXeGPU
 
-  DEPENDS
-  MLIRConversionPassIncGen
+    DEPENDS
+    MLIRConversionPassIncGen
 
-  LINK_LIBS PUBLIC
-  MLIRArithDialect
-  MLIRMemRefDialect
-  MLIRTransforms
-  MLIRVectorDialect
-  MLIRXeGPUDialect
-  )
+    LINK_LIBS PUBLIC
+    MLIRArithDialect
+    MLIRMemRefDialect
+    MLIRTransforms
+    MLIRVectorDialect
+    MLIRXeGPUDialect
+    )
+endif()

mlir/test/Conversion/VectorToXeGPU/load-to-xegpu.mlir

@@ -1,3 +1,4 @@
+// REQUIRES: vector-to-gpu-enabled
 // RUN: mlir-opt %s -convert-vector-to-xegpu -split-input-file | FileCheck %s
 
 func.func @load_1D_vector(%source: memref<8x16x32xf32>, %offset: index) -> vector<8xf32> {

mlir/test/Conversion/VectorToXeGPU/store-to-xegpu.mlir

@@ -1,3 +1,4 @@
+// REQUIRES: vector-to-gpu-enabled
 // RUN: mlir-opt %s -convert-vector-to-xegpu -split-input-file | FileCheck %s
 
 func.func @store_1D_vector(%vec: vector<8xf32>,

mlir/test/Conversion/VectorToXeGPU/transfer-read-to-xegpu.mlir

@@ -1,3 +1,4 @@
+// REQUIRES: vector-to-gpu-enabled
 // RUN: mlir-opt %s -convert-vector-to-xegpu -split-input-file | FileCheck %s
 
 func.func @load_1D_vector(%source: memref<8x16x32xf32>, %offset: index) -> vector<8xf32> {

mlir/test/Conversion/VectorToXeGPU/transfer-write-to-xegpu.mlir

@@ -1,3 +1,4 @@
+// REQUIRES: vector-to-gpu-enabled
 // RUN: mlir-opt %s -convert-vector-to-xegpu -split-input-file | FileCheck %s
 
 func.func @store_1D_vector(%vec: vector<8xf32>,

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

@@ -73,6 +73,10 @@ config.riscv_emulator_utils_lib_dir = "@RISCV_EMULATOR_UTILS_LIB_DIR@"
 if "@MLIR_CONVERSION_TOSATOTENSOR_ENABLE@" == "ON":
     config.available_features.add("tosa-to-tensor-enabled")
 
+# Add feature to enable/disable VectorToGPU tests
+if "@MLIR_DIALECT_XEGPU_ENABLE@" == "ON":
+    config.available_features.add("vector-to-gpu-enabled")
+
 import lit.llvm
 lit.llvm.initialize(lit_config, config)