File tree 4 files changed +52
-0
lines changed
4 files changed +52
-0
lines changed Original file line number Diff line number Diff line change @@ -434,6 +434,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
434
434
return UR_RESULT_SUCCESS;
435
435
}
436
436
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
+
437
450
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
438
451
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
439
452
auto result = validateProcInputs (version, pDdiTable);
Original file line number Diff line number Diff line change @@ -400,6 +400,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
400
400
return UR_RESULT_SUCCESS;
401
401
}
402
402
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
+
403
416
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
404
417
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
405
418
auto result = validateProcInputs (version, pDdiTable);
Original file line number Diff line number Diff line change @@ -418,6 +418,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
418
418
return UR_RESULT_SUCCESS;
419
419
}
420
420
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
+
421
434
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
422
435
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
423
436
auto result = validateProcInputs (version, pDdiTable);
Original file line number Diff line number Diff line change @@ -426,6 +426,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
426
426
return UR_RESULT_SUCCESS;
427
427
}
428
428
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
+
429
442
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
430
443
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
431
444
auto result = validateProcInputs (version, pDdiTable);
You can’t perform that action at this time.
0 commit comments