Skip to content

Commit 837aa27

Browse files
hdelannpmiller
authored andcommitted
Add ProcAddrTable Entry points
1 parent ccde31e commit 837aa27

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
@@ -434,6 +434,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
434434
return UR_RESULT_SUCCESS;
435435
}
436436

437+
UR_APIEXPORT ur_result_t UR_APICALL urGetTensorMapExpProcAddrTable(
438+
ur_api_version_t version, ur_tensor_map_exp_dditable_t *pDdiTable) {
439+
auto result = validateProcInputs(version, pDdiTable);
440+
if (UR_RESULT_SUCCESS != result) {
441+
return result;
442+
}
443+
444+
pDdiTable->pfnEncodeIm2ColExp = urTensorMapEncodeIm2ColExp;
445+
pDdiTable->pfnEncodeTiledExp = urTensorMapEncodeTiledExp;
446+
447+
return result;
448+
}
449+
437450
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable(
438451
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
439452
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
@@ -400,6 +400,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
400400
return UR_RESULT_SUCCESS;
401401
}
402402

403+
UR_APIEXPORT ur_result_t UR_APICALL urGetTensorMapExpProcAddrTable(
404+
ur_api_version_t version, ur_tensor_map_exp_dditable_t *pDdiTable) {
405+
auto result = validateProcInputs(version, pDdiTable);
406+
if (UR_RESULT_SUCCESS != result) {
407+
return result;
408+
}
409+
410+
pDdiTable->pfnEncodeIm2ColExp = urTensorMapEncodeIm2ColExp;
411+
pDdiTable->pfnEncodeTiledExp = urTensorMapEncodeTiledExp;
412+
413+
return result;
414+
}
415+
403416
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable(
404417
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
405418
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
@@ -418,6 +418,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
418418
return UR_RESULT_SUCCESS;
419419
}
420420

421+
UR_APIEXPORT ur_result_t UR_APICALL urGetTensorMapExpProcAddrTable(
422+
ur_api_version_t version, ur_tensor_map_exp_dditable_t *pDdiTable) {
423+
auto result = validateProcInputs(version, pDdiTable);
424+
if (UR_RESULT_SUCCESS != result) {
425+
return result;
426+
}
427+
428+
pDdiTable->pfnEncodeIm2ColExp = urTensorMapEncodeIm2ColExp;
429+
pDdiTable->pfnEncodeTiledExp = urTensorMapEncodeTiledExp;
430+
431+
return result;
432+
}
433+
421434
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable(
422435
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
423436
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
@@ -426,6 +426,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
426426
return UR_RESULT_SUCCESS;
427427
}
428428

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

0 commit comments

Comments
 (0)