File tree Expand file tree Collapse file tree 4 files changed +8
-6
lines changed Expand file tree Collapse file tree 4 files changed +8
-6
lines changed Original file line number Diff line number Diff line change @@ -196,13 +196,13 @@ typedef struct umf_memory_pool_ops_t {
196196 /// @brief Adds or removes devices on which allocations should be made
197197 /// resident.
198198 /// @param pool pointer to the memory pool
199- /// @param peerIdx identifier of device
199+ /// @param deviceIndex identifier of device
200200 /// @param isAdding Boolean indicating if peer is to be removed or added
201201 /// @return UMF_RESULT_SUCCESS on success or appropriate error code on
202202 /// failure.
203203 ///
204- umf_result_t (* ext_resident_device_change )(void * pool , uint32_t peerIdx ,
205- _Bool isAdding );
204+ umf_result_t (* ext_resident_device_change )(void * pool , uint32_t deviceIndex ,
205+ bool isAdding );
206206} umf_memory_pool_ops_t ;
207207
208208#ifdef __cplusplus
Original file line number Diff line number Diff line change @@ -54,7 +54,7 @@ UMF_1.0 {
5454 umfMemoryProviderPurgeForce;
5555 umfMemoryProviderPurgeLazy;
5656 umfMemoryProviderPutIPCHandle;
57- umfMemoryProviderResidentDeviceChange;
57+ umfMemoryProviderResidentDeviceChange;
5858 umfMempolicyCreate;
5959 umfMempolicyDestroy;
6060 umfMempolicySetCustomSplitPartitions;
Original file line number Diff line number Diff line change @@ -610,9 +610,12 @@ umf_result_t umfMemoryProviderGetAllocationPropertiesSize(
610610umf_result_t
611611umfMemoryProviderResidentDeviceChange (umf_memory_provider_handle_t hProvider ,
612612 uint32_t deviceIndex , bool isAdding ) {
613+
613614 UMF_CHECK ((hProvider != NULL ), UMF_RESULT_ERROR_INVALID_ARGUMENT );
615+
614616 umf_result_t res = hProvider -> ops .ext_resident_device_change (
615617 hProvider -> provider_priv , deviceIndex , isAdding );
618+
616619 checkErrorAndSetLastProvider (res , hProvider );
617620 return res ;
618621}
Original file line number Diff line number Diff line change 1717#include "memory_provider_internal.h"
1818#include "provider_ctl_stats_type.h"
1919#include "provider_level_zero_internal.h"
20+ #include "provider_tracking.h"
2021#include "utils_load_library.h"
2122#include "utils_log.h"
2223
23- #include <provider_tracking.h>
24-
2524static void * ze_lib_handle = NULL ;
2625
2726void fini_ze_global_state (void ) {
You can’t perform that action at this time.
0 commit comments