@@ -29,12 +29,10 @@ namespace ze_lib
2929 bool delayContextDestruction = false ;
3030 bool loaderTeardownCallbackReceived = false ;
3131 void staticLoaderTeardownCallback () {
32- printf (" ze_lib Static Teardown Callback\n " );
3332 loaderTeardownCallbackReceived = true ;
3433 }
3534 #endif
3635 void applicationTeardownCallback (uint32_t index) {
37- printf (" ze_lib Application Teardown Callback %d\n " , index);
3836 if (ze_lib::context->teardownCallbacks .find (index) != ze_lib::context->teardownCallbacks .end ()) {
3937 ze_lib::context->teardownCallbacks .erase (index);
4038 }
@@ -50,9 +48,6 @@ namespace ze_lib
5048 // /////////////////////////////////////////////////////////////////////////////
5149 __zedlllocal context_t ::~context_t ()
5250 {
53- if (debugTraceEnabled) {
54- debug_trace_message (" ze_lib Context Destructor" , " " );
55- }
5651#ifdef DYNAMIC_LOAD_LOADER
5752 if (!loaderTeardownCallbackReceived) {
5853 loaderTeardownCallback (loaderTeardownCallbackIndex);
@@ -372,8 +367,8 @@ namespace ze_lib
372367 }
373368 // Get the function pointer for zelRegisterTeardownCallback from the loader
374369 typedef ze_result_t (ZE_APICALL *zelRegisterTeardownCallback_t)(
375- zel_loader_teardown_callback_t ,
376- zel_application_teardown_callback_t *,
370+ zel_loader_teardown_callback_t ,
371+ zel_application_teardown_callback_t *,
377372 uint32_t *);
378373 auto pfnZelRegisterTeardownCallback = reinterpret_cast <zelRegisterTeardownCallback_t>(
379374 GET_FUNCTION_PTR (loader, " zelRegisterTeardownCallback" ));
@@ -518,11 +513,11 @@ void stabilityCheck(std::promise<int> stabilityPromise) {
518513}
519514#endif
520515
521- ZE_DLLEXPORT ze_result_t ZE_APICALL
516+ ze_result_t ZE_APICALL
522517zelRegisterTeardownCallback (
523- zel_loader_teardown_callback_t application_callback, // [in] Pointer to the user's application callback function
524- zel_application_teardown_callback_t *loader_callback, // [out] Pointer to the L0 Loader 's callback function
525- uint32_t *index // [out] Index of the callback function
518+ zel_loader_teardown_callback_t application_callback, // [in] Application's callback function to be called during loader teardown
519+ zel_application_teardown_callback_t *loader_callback, // [out] Pointer to the loader 's callback function
520+ uint32_t *index // [out] Index assigned to the registered callback
526521) {
527522 ze_result_t result = ZE_RESULT_SUCCESS ;
528523 if (nullptr == application_callback) {
0 commit comments