@@ -81,9 +81,10 @@ static pi_result getExtFuncFromContext(pi_context context, T *fptr) {
81
81
return PI_SUCCESS;
82
82
}
83
83
84
- size_t deviceCount;
85
- cl_int ret_err = clGetContextInfo (
86
- cast<cl_context>(context), CL_CONTEXT_DEVICES, 0 , nullptr , &deviceCount);
84
+ cl_uint deviceCount;
85
+ cl_int ret_err =
86
+ clGetContextInfo (cast<cl_context>(context), CL_CONTEXT_NUM_DEVICES,
87
+ sizeof (cl_uint), &deviceCount, nullptr );
87
88
88
89
if (ret_err != CL_SUCCESS || deviceCount < 1 ) {
89
90
return PI_INVALID_CONTEXT;
@@ -354,11 +355,10 @@ pi_result piextQueueCreateWithNativeHandle(pi_native_handle nativeHandle,
354
355
355
356
pi_result piProgramCreate (pi_context context, const void *il, size_t length,
356
357
pi_program *res_program) {
357
-
358
- size_t deviceCount;
359
-
360
- cl_int ret_err = clGetContextInfo (
361
- cast<cl_context>(context), CL_CONTEXT_DEVICES, 0 , nullptr , &deviceCount);
358
+ cl_uint deviceCount;
359
+ cl_int ret_err =
360
+ clGetContextInfo (cast<cl_context>(context), CL_CONTEXT_NUM_DEVICES,
361
+ sizeof (cl_uint), &deviceCount, nullptr );
362
362
363
363
std::vector<cl_device_id> devicesInCtx (deviceCount);
364
364
0 commit comments