fix tests

Signed-off-by: dchigarev <[email protected]>
intel · Aug 1, 2024 · f78f6d2 · f78f6d2
1 parent c8ff254
commit f78f6d2
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/cmake/imex.cmake b/cmake/imex.cmake
@@ -7,7 +7,7 @@ if (NOT DEFINED IMEX_INCLUDES)
     # TODO: Change to main https://github.com/oneapi-src/oneDNN.git when all the
     # required functionality is merged.
     gc_fetch_content(imex 496b240093b5e132b60c5ee69878300fe69be300 https://github.com/Menooker/mlir-extensions
-            CMAKE_ARGS "-DMLIR_DIR=${MLIR_DIR};-DIMEX_CHECK_LLVM_VERSION=ON"
+            CMAKE_ARGS "-DMLIR_DIR=${MLIR_DIR};-DIMEX_CHECK_LLVM_VERSION=ON;-DIMEX_ENABLE_L0_RUNTIME=${IMEX_ENABLE_L0_RUNTIME}"
     )
 
     set(IMEX_INCLUDES

diff --git a/test/mlir/test/CMakeLists.txt b/test/mlir/test/CMakeLists.txt
@@ -26,6 +26,9 @@ set(GC_OPT_TEST_DEPENDS
 if(GC_USE_GPU)
         include(imex)
         list(APPEND GC_OPT_TEST_DEPENDS mlir_opencl_runtime)
+        if (IMEX_ENABLE_L0_RUNTIME)
+                list(APPEND GC_OPT_TEST_DEPENDS level-zero-runtime)
+        endif()
 endif()
 
 if(GC_ENABLE_BINDINGS_PYTHON)