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 @@ -433,6 +433,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
433
433
return UR_RESULT_SUCCESS;
434
434
}
435
435
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
+
436
449
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
437
450
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
438
451
auto result = validateProcInputs (version, pDdiTable);
Original file line number Diff line number Diff line change @@ -399,6 +399,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
399
399
return UR_RESULT_SUCCESS;
400
400
}
401
401
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
+
402
415
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
403
416
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
404
417
auto result = validateProcInputs (version, pDdiTable);
Original file line number Diff line number Diff line change @@ -417,6 +417,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
417
417
return UR_RESULT_SUCCESS;
418
418
}
419
419
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
+
420
433
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
421
434
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
422
435
auto result = validateProcInputs (version, pDdiTable);
Original file line number Diff line number Diff line change @@ -424,6 +424,19 @@ UR_DLLEXPORT ur_result_t UR_APICALL urGetKernelExpProcAddrTable(
424
424
return UR_RESULT_SUCCESS;
425
425
}
426
426
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
+
427
440
UR_DLLEXPORT ur_result_t UR_APICALL urGetProgramExpProcAddrTable (
428
441
ur_api_version_t version, ur_program_exp_dditable_t *pDdiTable) {
429
442
auto result = validateProcInputs (version, pDdiTable);
You can’t perform that action at this time.
0 commit comments