File tree Expand file tree Collapse file tree 1 file changed +10
-6
lines changed Expand file tree Collapse file tree 1 file changed +10
-6
lines changed Original file line number Diff line number Diff line change @@ -38,15 +38,19 @@ int ur_device_get(TestState &state) {
38
38
return -1 ;
39
39
}
40
40
41
- ur_result_t res = urDeviceGet (state.platforms [state.platform_num ],
42
- state.device_type , state.num_entries ,
43
- state.devices .data (), &state.num_devices );
41
+ ur_result_t res = UR_RESULT_SUCCESS;
42
+ if (state.devices .size () == 0 ) {
43
+ res = urDeviceGet (state.platforms [state.platform_num ],
44
+ state.device_type , 0 , nullptr , &state.num_devices );
45
+ state.devices .resize (state.num_devices );
46
+ } else {
47
+ res =
48
+ urDeviceGet (state.platforms [state.platform_num ], state.device_type ,
49
+ state.num_entries , state.devices .data (), nullptr );
50
+ }
44
51
if (res != UR_RESULT_SUCCESS) {
45
52
return -1 ;
46
53
}
47
- if (state.devices .size () != state.num_devices ) {
48
- state.devices .resize (state.num_devices );
49
- }
50
54
51
55
return 0 ;
52
56
}
You can’t perform that action at this time.
0 commit comments