Skip to content

Commit df93280

Browse files
committed
Add ProcAddrTable Entry points
1 parent a993497 commit df93280

File tree

4 files changed

+52
-0
lines changed

4 files changed

+52
-0
lines changed

source/adapters/cuda/ur_interface_loader.cpp

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -433,6 +433,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
433433
return UR_RESULT_SUCCESS;
434434
}
435435

436+
UR_APIEXPORT ur_result_t UR_APICALL urGetTensorMapExpProcAddrTable(
437+
ur_api_version_t version, ur_tensor_map_exp_dditable_t *pDdiTable) {
438+
auto result = validateProcInputs(version, pDdiTable);
439+
if (UR_RESULT_SUCCESS != result) {
440+
return result;
441+
}
442+
443+
pDdiTable->pfnEncodeIm2ColExp = urTensorMapEncodeIm2ColExp;
444+
pDdiTable->pfnEncodeTiledExp = urTensorMapEncodeTiledExp;
445+
446+
return result;
447+
}
448+
436449
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable(
437450
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
438451
auto result = validateProcInputs(version, pDdiTable);

source/adapters/hip/ur_interface_loader.cpp

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -399,6 +399,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
399399
return UR_RESULT_SUCCESS;
400400
}
401401

402+
UR_APIEXPORT ur_result_t UR_APICALL urGetTensorMapExpProcAddrTable(
403+
ur_api_version_t version, ur_tensor_map_exp_dditable_t *pDdiTable) {
404+
auto result = validateProcInputs(version, pDdiTable);
405+
if (UR_RESULT_SUCCESS != result) {
406+
return result;
407+
}
408+
409+
pDdiTable->pfnEncodeIm2ColExp = urTensorMapEncodeIm2ColExp;
410+
pDdiTable->pfnEncodeTiledExp = urTensorMapEncodeTiledExp;
411+
412+
return result;
413+
}
414+
402415
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable(
403416
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
404417
auto result = validateProcInputs(version, pDdiTable);

source/adapters/native_cpu/ur_interface_loader.cpp

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -417,6 +417,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
417417
return UR_RESULT_SUCCESS;
418418
}
419419

420+
UR_APIEXPORT ur_result_t UR_APICALL urGetTensorMapExpProcAddrTable(
421+
ur_api_version_t version, ur_tensor_map_exp_dditable_t *pDdiTable) {
422+
auto result = validateProcInputs(version, pDdiTable);
423+
if (UR_RESULT_SUCCESS != result) {
424+
return result;
425+
}
426+
427+
pDdiTable->pfnEncodeIm2ColExp = urTensorMapEncodeIm2ColExp;
428+
pDdiTable->pfnEncodeTiledExp = urTensorMapEncodeTiledExp;
429+
430+
return result;
431+
}
432+
420433
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable(
421434
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
422435
auto result = validateProcInputs(version, pDdiTable);

source/adapters/opencl/ur_interface_loader.cpp

Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -424,6 +424,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
424424
return UR_RESULT_SUCCESS;
425425
}
426426

427+
UR_APIEXPORT ur_result_t UR_APICALL urGetTensorMapExpProcAddrTable(
428+
ur_api_version_t version, ur_tensor_map_exp_dditable_t *pDdiTable) {
429+
auto result = validateProcInputs(version, pDdiTable);
430+
if (UR_RESULT_SUCCESS != result) {
431+
return result;
432+
}
433+
434+
pDdiTable->pfnEncodeIm2ColExp = urTensorMapEncodeIm2ColExp;
435+
pDdiTable->pfnEncodeTiledExp = urTensorMapEncodeTiledExp;
436+
437+
return result;
438+
}
439+
427440
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable(
428441
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
429442
auto result = validateProcInputs(version, pDdiTable);

0 commit comments

Comments
 (0)