@@ -83,7 +83,7 @@ inline ur_result_t mock_urDeviceGet(void *pParams) {
83
83
inline ur_result_t mock_urDeviceRetain (void *) { return UR_RESULT_SUCCESS; }
84
84
inline ur_result_t mock_urDeviceRelease (void *) { return UR_RESULT_SUCCESS; }
85
85
86
- template <ur_backend_t Backend>
86
+ template <ur_adapter_backend_t Backend>
87
87
inline ur_result_t mock_urAdapterGetInfo (void *pParams) {
88
88
auto params = reinterpret_cast <ur_adapter_get_info_params_t *>(pParams);
89
89
@@ -100,7 +100,7 @@ inline ur_result_t mock_urAdapterGetInfo(void *pParams) {
100
100
return UR_RESULT_SUCCESS;
101
101
}
102
102
103
- template <ur_backend_t Backend>
103
+ template <ur_platform_backend_t Backend>
104
104
inline ur_result_t mock_urPlatformGetInfo (void *pParams) {
105
105
auto params = reinterpret_cast <ur_platform_get_info_params_t *>(pParams);
106
106
constexpr char MockPlatformName[] = " Mock platform" ;
@@ -556,14 +556,16 @@ template <sycl::backend Backend = backend::opencl> class UrMock {
556
556
#define ADD_DEFAULT_OVERRIDE (func_name, func_override ) \
557
557
mock::getCallbacks ().set_replace_callback (#func_name, \
558
558
&MockAdapter::func_override);
559
- ADD_DEFAULT_OVERRIDE (urAdapterGetInfo,
560
- mock_urAdapterGetInfo<convertToUrBackend (Backend)>)
559
+ ADD_DEFAULT_OVERRIDE (
560
+ urAdapterGetInfo,
561
+ mock_urAdapterGetInfo<convertToUrAdapterBackend (Backend)>)
561
562
ADD_DEFAULT_OVERRIDE (urPlatformGet, mock_urPlatformGet)
562
563
ADD_DEFAULT_OVERRIDE (urDeviceGet, mock_urDeviceGet)
563
564
ADD_DEFAULT_OVERRIDE (urDeviceRetain, mock_urDeviceRetain)
564
565
ADD_DEFAULT_OVERRIDE (urDeviceRelease, mock_urDeviceRelease)
565
- ADD_DEFAULT_OVERRIDE (urPlatformGetInfo,
566
- mock_urPlatformGetInfo<convertToUrBackend (Backend)>)
566
+ ADD_DEFAULT_OVERRIDE (
567
+ urPlatformGetInfo,
568
+ mock_urPlatformGetInfo<convertToUrPlatformBackend (Backend)>)
567
569
ADD_DEFAULT_OVERRIDE (urDeviceGetInfo, mock_urDeviceGetInfo)
568
570
ADD_DEFAULT_OVERRIDE (urProgramGetInfo, mock_urProgramGetInfo)
569
571
ADD_DEFAULT_OVERRIDE (urContextGetInfo, mock_urContextGetInfo)
@@ -617,21 +619,39 @@ template <sycl::backend Backend = backend::opencl> class UrMock {
617
619
private:
618
620
// These two helpers are needed to enable arbitrary backend selection
619
621
// at compile time.
620
- static constexpr ur_backend_t
621
- convertToUrBackend (const sycl::backend SyclBackend) {
622
+ static constexpr ur_platform_backend_t
623
+ convertToUrPlatformBackend (const sycl::backend SyclBackend) {
622
624
switch (SyclBackend) {
623
625
case sycl::backend::opencl:
624
- return UR_BACKEND_OPENCL ;
626
+ return UR_PLATFORM_BACKEND_OPENCL ;
625
627
case sycl::backend::ext_oneapi_level_zero:
626
- return UR_BACKEND_LEVEL_ZERO ;
628
+ return UR_PLATFORM_BACKEND_LEVEL_ZERO ;
627
629
case sycl::backend::ext_oneapi_cuda:
628
- return UR_BACKEND_CUDA ;
630
+ return UR_PLATFORM_BACKEND_CUDA ;
629
631
case sycl::backend::ext_oneapi_hip:
630
- return UR_BACKEND_HIP ;
632
+ return UR_PLATFORM_BACKEND_HIP ;
631
633
case sycl::backend::ext_oneapi_native_cpu:
632
- return UR_BACKEND_NATIVE_CPU ;
634
+ return UR_PLATFORM_BACKEND_NATIVE_CPU ;
633
635
default :
634
- return UR_BACKEND_UNKNOWN;
636
+ return UR_PLATFORM_BACKEND_UNKNOWN;
637
+ }
638
+ }
639
+
640
+ static constexpr ur_adapter_backend_t
641
+ convertToUrAdapterBackend (sycl::backend SyclBackend) {
642
+ switch (SyclBackend) {
643
+ case sycl::backend::opencl:
644
+ return UR_ADAPTER_BACKEND_OPENCL;
645
+ case sycl::backend::ext_oneapi_level_zero:
646
+ return UR_ADAPTER_BACKEND_LEVEL_ZERO;
647
+ case sycl::backend::ext_oneapi_cuda:
648
+ return UR_ADAPTER_BACKEND_CUDA;
649
+ case sycl::backend::ext_oneapi_hip:
650
+ return UR_ADAPTER_BACKEND_HIP;
651
+ case sycl::backend::ext_oneapi_native_cpu:
652
+ return UR_ADAPTER_BACKEND_NATIVE_CPU;
653
+ default :
654
+ return UR_ADAPTER_BACKEND_UNKNOWN;
635
655
}
636
656
}
637
657
};
0 commit comments