Merge pull request #947 from ldorau/Move_implementation_from_examples_level_zero.h_to_examples_level_zero.c

Move implementation from `examples_level_zero.h` to `examples_level_zero.c`

Author: ldorau

examples/CMakeLists.txt

@@ -49,6 +49,7 @@ if(UMF_BUILD_GPU_EXAMPLES
     add_umf_executable(
         NAME ${EXAMPLE_NAME}
         SRCS level_zero_shared_memory/level_zero_shared_memory.c
+             common/examples_level_zero_helpers.c
         LIBS disjoint_pool ze_loader umf)
 
     target_include_directories(
@@ -127,6 +128,7 @@ if(UMF_BUILD_GPU_EXAMPLES
     add_umf_executable(
         NAME ${EXAMPLE_NAME}
         SRCS ipc_level_zero/ipc_level_zero.c
+             common/examples_level_zero_helpers.c
         LIBS disjoint_pool ze_loader umf)
 
     target_include_directories(

examples/common/examples_level_zero.h renamed to examples/common/examples_level_zero_helpers.c

@@ -7,21 +7,12 @@
  *
  */
 
-#ifndef UMF_EXAMPLE_UTILS_LEVEL_ZERO_H
-#define UMF_EXAMPLE_UTILS_LEVEL_ZERO_H
-
 #include <stdio.h>
 #include <stdlib.h>
 
-// To use the Level Zero API, the Level Zero SDK has to be installed
-// on the system
-#ifdef _WIN32
-#include <ze_api.h>
-#else
-#include <level_zero/ze_api.h>
-#endif
+#include "examples_level_zero_helpers.h"
 
-static int init_level_zero(void) {
+int init_level_zero(void) {
     ze_init_flag_t flags = ZE_INIT_FLAG_GPU_ONLY;
     ze_result_t result = zeInit(flags);
     if (result != ZE_RESULT_SUCCESS) {
@@ -118,8 +109,7 @@ static inline int get_devices(ze_driver_handle_t driver, uint32_t *devices_num_,
     return ret;
 }
 
-static inline int find_driver_with_gpu(uint32_t *driver_idx,
-                                       ze_driver_handle_t *driver_) {
+int find_driver_with_gpu(uint32_t *driver_idx, ze_driver_handle_t *driver_) {
     int ret = 0;
     ze_result_t ze_result;
     uint32_t drivers_num = 0;
@@ -184,8 +174,7 @@ static inline int find_driver_with_gpu(uint32_t *driver_idx,
     return ret;
 }
 
-static inline int find_gpu_device(ze_driver_handle_t driver,
-                                  ze_device_handle_t *device_) {
+int find_gpu_device(ze_driver_handle_t driver, ze_device_handle_t *device_) {
     int ret = -1;
     uint32_t devices_num = 0;
     ze_device_handle_t *devices = NULL;
@@ -415,5 +404,3 @@ int destroy_context(ze_context_handle_t context) {
 
     return 0;
 }
-
-#endif // UMF_EXAMPLE_UTILS_LEVEL_ZERO_H

examples/common/examples_level_zero_helpers.h

@@ -0,0 +1,38 @@
+/*
+ *
+ * Copyright (C) 2024 Intel Corporation
+ *
+ * Under the Apache License v2.0 with LLVM Exceptions. See LICENSE.TXT.
+ * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+ *
+ */
+
+#ifndef UMF_EXAMPLES_LEVEL_ZERO_H
+#define UMF_EXAMPLES_LEVEL_ZERO_H
+
+// To use the Level Zero API, the Level Zero SDK has to be installed
+// on the system
+#ifdef _WIN32
+#include <ze_api.h>
+#else
+#include <level_zero/ze_api.h>
+#endif
+
+int init_level_zero(void);
+
+int create_context(ze_driver_handle_t driver, ze_context_handle_t *context);
+
+int destroy_context(ze_context_handle_t context);
+
+int find_driver_with_gpu(uint32_t *driver_idx, ze_driver_handle_t *driver_);
+
+int find_gpu_device(ze_driver_handle_t driver, ze_device_handle_t *device_);
+
+int level_zero_fill(ze_context_handle_t context, ze_device_handle_t device,
+                    void *ptr, size_t size, const void *pattern,
+                    size_t pattern_size);
+
+int level_zero_copy(ze_context_handle_t context, ze_device_handle_t device,
+                    void *dst_ptr, void *src_ptr, size_t size);
+
+#endif /* UMF_EXAMPLES_LEVEL_ZERO_H */

examples/ipc_level_zero/CMakeLists.txt

@@ -45,7 +45,9 @@ message(STATUS "Level Zero include directory: ${LEVEL_ZERO_INCLUDE_DIRS}")
 
 # build the example
 set(EXAMPLE_NAME umf_example_ipc_level_zero)
-add_executable(${EXAMPLE_NAME} ipc_level_zero.c)
+add_executable(
+    ${EXAMPLE_NAME} ipc_level_zero.c
+                    ${UMF_EXAMPLE_DIR}/common/examples_level_zero_helpers.c)
 target_include_directories(${EXAMPLE_NAME} PRIVATE ${LIBUMF_INCLUDE_DIRS}
                                                    ${UMF_EXAMPLE_DIR}/common)
 target_link_directories(${EXAMPLE_NAME} PRIVATE ${LIBUMF_LIBRARY_DIRS}

examples/ipc_level_zero/ipc_level_zero.c

@@ -8,13 +8,14 @@
  */
 
 #include <stdio.h>
+#include <stdlib.h>
 
 #include "umf/ipc.h"
 #include "umf/memory_pool.h"
 #include "umf/pools/pool_disjoint.h"
 #include "umf/providers/provider_level_zero.h"
 
-#include "examples_level_zero.h"
+#include "examples_level_zero_helpers.h"
 
 int create_level_zero_pool(ze_context_handle_t context,
                            ze_device_handle_t device,

examples/level_zero_shared_memory/CMakeLists.txt

@@ -45,7 +45,9 @@ message(STATUS "Level Zero include directory: ${LEVEL_ZERO_INCLUDE_DIRS}")
 
 # build the example
 set(EXAMPLE_NAME umf_example_level_zero_shared_memory)
-add_executable(${EXAMPLE_NAME} level_zero_shared_memory.c)
+add_executable(
+    ${EXAMPLE_NAME} level_zero_shared_memory.c
+                    ${UMF_EXAMPLE_DIR}/common/examples_level_zero_helpers.c)
 target_include_directories(${EXAMPLE_NAME} PRIVATE ${LIBUMF_INCLUDE_DIRS}
                                                    ${UMF_EXAMPLE_DIR}/common)
 target_link_directories(${EXAMPLE_NAME} PRIVATE ${LIBUMF_LIBRARY_DIRS}

examples/level_zero_shared_memory/level_zero_shared_memory.c

@@ -7,11 +7,13 @@
  *
  */
 
+#include <stdio.h>
+
 #include <umf/memory_pool.h>
 #include <umf/pools/pool_disjoint.h>
 #include <umf/providers/provider_level_zero.h>
 
-#include "examples_level_zero.h"
+#include "examples_level_zero_helpers.h"
 
 int main(void) {
     // A result object for storing UMF API result status