File tree Expand file tree Collapse file tree 2 files changed +18
-5
lines changed Expand file tree Collapse file tree 2 files changed +18
-5
lines changed Original file line number Diff line number Diff line change @@ -144,10 +144,15 @@ int InitLevelZeroOps() {
144
144
const char *lib_name = " libze_loader.so" ;
145
145
#endif
146
146
// Load Level Zero symbols
147
- // NOTE that we use UMF_UTIL_OPEN_LIBRARY_GLOBAL which add all loaded symbols to the
147
+ #if OPEN_ZE_LIBRARY_GLOBAL
148
+ // NOTE UMF_UTIL_OPEN_LIBRARY_GLOBAL adds all loaded symbols to the
148
149
// global symbol table.
150
+ int open_flags = UMF_UTIL_OPEN_LIBRARY_GLOBAL;
151
+ #else
152
+ int open_flags = 0 ;
153
+ #endif
149
154
zeDlHandle = std::unique_ptr<void , DlHandleCloser>(
150
- utils_open_library (lib_name, UMF_UTIL_OPEN_LIBRARY_GLOBAL ));
155
+ utils_open_library (lib_name, open_flags ));
151
156
*(void **)&libze_ops.zeInit =
152
157
utils_get_symbol_addr (zeDlHandle.get (), " zeInit" , lib_name);
153
158
if (libze_ops.zeInit == nullptr ) {
Original file line number Diff line number Diff line change @@ -415,12 +415,20 @@ if(UMF_BUILD_GPU_TESTS AND UMF_BUILD_LEVEL_ZERO_PROVIDER)
415
415
LIBS ${UMF_UTILS_FOR_TEST} ze_loader )
416
416
417
417
add_umf_test (
418
- NAME provider_level_zero_dlopen
418
+ NAME provider_level_zero_dlopen_global
419
419
SRCS providers/provider_level_zero.cpp
420
420
${UMF_UTILS_DIR} /utils_level_zero.cpp ${BA_SOURCES_FOR_TEST}
421
421
LIBS ${UMF_UTILS_FOR_TEST} )
422
- target_compile_definitions (umf_test-provider_level_zero_dlopen
423
- PUBLIC USE_DLOPEN=1 )
422
+ target_compile_definitions (umf_test-provider_level_zero_dlopen_global
423
+ PUBLIC USE_DLOPEN=1 OPEN_ZE_LIBRARY_GLOBAL=1 )
424
+
425
+ add_umf_test (
426
+ NAME provider_level_zero_dlopen_local
427
+ SRCS providers/provider_level_zero.cpp
428
+ ${UMF_UTILS_DIR} /utils_level_zero.cpp ${BA_SOURCES_FOR_TEST}
429
+ LIBS ${UMF_UTILS_FOR_TEST} )
430
+ target_compile_definitions (umf_test-provider_level_zero_dlopen_local
431
+ PUBLIC USE_DLOPEN=1 OPEN_ZE_LIBRARY_GLOBAL=0 )
424
432
endif ()
425
433
426
434
if (NOT UMF_BUILD_LEVEL_ZERO_PROVIDER )
You can’t perform that action at this time.
0 commit comments