Skip to content

Commit 9367aac

Browse files
omarahmed1111callumfare
authored andcommitted
[SYCL][CUDA][UR] Add usmPool entry points to ddi tables and fix ur*nativeHandle apis
1 parent f7a9f28 commit 9367aac

File tree

4 files changed

+8
-4
lines changed

4 files changed

+8
-4
lines changed

sycl/plugins/unified_runtime/ur/adapters/cuda/device.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1078,6 +1078,7 @@ UR_APIEXPORT ur_result_t UR_APICALL urDeviceGetNativeHandle(
10781078

10791079
UR_APIEXPORT ur_result_t UR_APICALL urDeviceCreateWithNativeHandle(
10801080
ur_native_handle_t hNativeDevice, ur_platform_handle_t hPlatform,
1081+
const ur_device_native_properties_t *pProperties,
10811082
ur_device_handle_t *phDevice) {
10821083
UR_ASSERT(phDevice, UR_RESULT_ERROR_INVALID_NULL_POINTER);
10831084

sycl/plugins/unified_runtime/ur/adapters/cuda/program.cpp

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -281,6 +281,7 @@ urProgramLink(ur_context_handle_t hContext, uint32_t count,
281281
/// \return TBD
282282
UR_APIEXPORT ur_result_t UR_APICALL urProgramCreateWithNativeHandle(
283283
ur_native_handle_t hNativeProgram, ur_context_handle_t hContext,
284+
const ur_program_native_properties_t *pProperties,
284285
ur_program_handle_t *phProgram) {
285286
return UR_RESULT_ERROR_UNSUPPORTED_FEATURE;
286287
}

sycl/plugins/unified_runtime/ur/adapters/cuda/queue.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -236,8 +236,9 @@ UR_APIEXPORT ur_result_t UR_APICALL urQueueFlush(ur_queue_handle_t hQueue) {
236236
return UR_RESULT_SUCCESS;
237237
}
238238

239-
UR_APIEXPORT ur_result_t UR_APICALL urQueueGetNativeHandle(
240-
ur_queue_handle_t hQueue, ur_native_handle_t *phNativeQueue) {
239+
UR_APIEXPORT ur_result_t UR_APICALL
240+
urQueueGetNativeHandle(ur_queue_handle_t hQueue, ur_queue_native_desc_t *pDesc,
241+
ur_native_handle_t *phNativeQueue) {
241242
UR_ASSERT(hQueue, UR_RESULT_ERROR_INVALID_NULL_HANDLE);
242243
UR_ASSERT(phNativeQueue, UR_RESULT_ERROR_INVALID_NULL_POINTER);
243244

sycl/plugins/unified_runtime/ur/adapters/cuda/ur_interface_loader.cpp

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -233,8 +233,9 @@ urGetUSMProcAddrTable(ur_api_version_t version, ur_usm_dditable_t *pDdiTable) {
233233
pDdiTable->pfnGetMemAllocInfo = urUSMGetMemAllocInfo;
234234
pDdiTable->pfnHostAlloc = urUSMHostAlloc;
235235
pDdiTable->pfnPoolCreate = nullptr;
236-
pDdiTable->pfnPoolDestroy = nullptr;
237-
pDdiTable->pfnPoolDestroy = nullptr;
236+
pDdiTable->pfnPoolRetain = nullptr;
237+
pDdiTable->pfnPoolRelease = nullptr;
238+
pDdiTable->pfnPoolGetInfo = nullptr;
238239
pDdiTable->pfnSharedAlloc = urUSMSharedAlloc;
239240
return UR_RESULT_SUCCESS;
240241
}

0 commit comments

Comments
 (0)