Added GC_ENABLE_IMEX option

Author: AndreyPavlenko

CMakeLists.txt

@@ -33,6 +33,7 @@ option(GC_ENABLE_TEST_DNNL "Build the dnnl tests" ${GC_ENABLE_DNNL})
 option(GC_ENABLE_TEST_MLIR "Build the mlir tests" ON)
 option(GC_ENABLE_OPT "Build gc-opt" ON)
 option(GC_ENABLE_GPU "Enable GPU backend" OFF)
+option(GC_ENABLE_IMEX "Intel® Extension for MLIR" OFF)
 option(GC_ENABLE_BINDINGS_PYTHON "Enable Graph Complier Python Binding" ON)
 option(GC_DEV_LINK_LLVM_DYLIB "Link dynamic libraries of LLVM and MLIR. For developers only. Do not use it in packing the library." OFF)
 
@@ -63,6 +64,10 @@ if(GC_ENABLE_DNNL)
 endif ()
 
 if(GC_ENABLE_GPU)
+  add_definitions(-DGC_USE_GPU)
+endif ()
+
+if(GC_ENABLE_IMEX)
   include(imex)
   if(GC_DEV_LINK_LLVM_DYLIB)
     message(WARN "GPU backend may not be compatible with dynamic linking to LLVM")

src/gc-opt/CMakeLists.txt

@@ -40,8 +40,8 @@ set(gc_opt_libs
         ${MLIR_LINK_COMPONENTS}
         GCPasses)
 
-if(GC_ENABLE_GPU)
-  add_definitions(-DGC_USE_GPU=1)
+if(GC_ENABLE_IMEX)
+  add_definitions(-DGC_USE_IMEX)
   get_property(IMEX_INCLUDES GLOBAL PROPERTY IMEX_INCLUDES)
   include_directories(${IMEX_INCLUDES})
   list(APPEND gc_opt_libs IMEXGPUXDialect IMEXXeTileDialect IMEXRegionDialect IMEXRegionTransforms

src/gc-opt/gc-opt.cpp

@@ -28,7 +28,7 @@
 #include "mlir/InitAllPasses.h"
 #include "mlir/Tools/mlir-opt/MlirOptMain.h"
 
-#ifdef GC_USE_GPU
+#ifdef GC_USE_IMEX
 #include <imex/InitIMEXDialects.h>
 #include <imex/InitIMEXPasses.h>
 #endif
@@ -38,7 +38,7 @@ void registerCPUPipeline();
 } // namespace mlir::gc
 
 int main(int argc, char *argv[]) {
-#ifdef GC_USE_GPU
+#ifdef GC_USE_IMEX
   imex::registerTransformsPasses();
   // Conversion passes
   imex::registerConvertGPUToGPUX();
@@ -59,7 +59,7 @@ int main(int argc, char *argv[]) {
   registry.insert<mlir::linalgx::LinalgxDialect>();
   registry.insert<mlir::microkernel::MicrokernelDialect>();
   mlir::registerAllDialects(registry);
-#ifdef GC_USE_GPU
+#ifdef GC_USE_IMEX
   registry.insert<::imex::xetile::XeTileDialect, ::imex::gpux::GPUXDialect>();
 #endif
   mlir::cpuruntime::registerConvertCPURuntimeToLLVMInterface(registry);

test/mlir/test/CMakeLists.txt

@@ -23,7 +23,7 @@ set(GC_OPT_TEST_DEPENDS
         GCUnitTests
 )
 
-if(GC_ENABLE_GPU)
+if(GC_ENABLE_IMEX)
         include(imex)
         list(APPEND GC_OPT_TEST_DEPENDS mlir_opencl_runtime)
 endif()

test/mlir/test/gc/gpu-runner/lit.local.cfg

@@ -1,2 +1,2 @@
-if not config.gc_use_gpu:
+if not config.gc_use_imex:
     config.unsupported = True

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

@@ -34,7 +34,7 @@ config.gc_src_root = "@PROJECT_SOURCE_DIR@"
 config.gc_obj_root = "@PROJECT_BINARY_DIR@"
 config.gc_lib_dir = os.path.join(config.gc_obj_root, "lib")
 config.mlir_obj_dir = "@MLIR_BINARY_DIR@"
-config.gc_use_gpu = "@GC_ENABLE_GPU@" in ["ON", "1"]
+config.gc_use_imex = "@GC_ENABLE_IMEX@" in ["ON", "1"]
 config.enable_bindings_python = @GC_ENABLE_BINDINGS_PYTHON@
 config.llvm_shlib_dir = "@SHLIBDIR@"
 config.llvm_shlib_ext = "@SHLIBEXT@"