void cmd_device(struct console* console, int argc, const char** argv)
{
struct device_info info;
rcl_status status;
int i;
int dims;
(void)argc;
(void)argv;
status = cl_get_device_info(&info);
if (status) {
console_print(console, "Error obtaining device info: %lu\n",
(unsigned long)status);
return;
}
console_print(console, "Global memory:\t\t%" PRIu64 " bytes\n" \
"Local memory:\t\t%" PRIu64 " bytes\nCompute units:\t\t%" PRIu32 \
"\nWork group size:\t%" PRIu32 "\nWork item dimensions:\t%" PRIu32 \
"\n", info.global_memory, info.local_memory, info.compute_units,
info.work_group_size, info.work_item_sizes.work_item_sizes_len);
dims = info.work_item_sizes.work_item_sizes_len;
console_print(console, "Work item sizes:\t");
for (i = 0; i < dims; i++) {
console_print(console, "%" PRIu32 "%s",
info.work_item_sizes.work_item_sizes_val[i],
i + 1 < dims ? ", " : "\n");
}
free(info.work_item_sizes.work_item_sizes_val);
}
LOCAL cl_mem
cl_mem_new_buffer(cl_context ctx,
cl_mem_flags flags,
size_t sz,
void *data,
cl_int *errcode_ret)
{
/* Possible mem type combination:
CL_MEM_ALLOC_HOST_PTR
CL_MEM_ALLOC_HOST_PTR | CL_MEM_COPY_HOST_PTR
CL_MEM_USE_HOST_PTR
CL_MEM_COPY_HOST_PTR */
cl_int err = CL_SUCCESS;
cl_mem mem = NULL;
cl_ulong max_mem_size;
if (UNLIKELY(sz == 0)) {
err = CL_INVALID_BUFFER_SIZE;
goto error;
}
if (UNLIKELY(((flags & CL_MEM_READ_WRITE)
&& (flags & (CL_MEM_READ_ONLY | CL_MEM_WRITE_ONLY)))
|| ((flags & CL_MEM_READ_ONLY) && (flags & (CL_MEM_WRITE_ONLY)))
|| ((flags & CL_MEM_ALLOC_HOST_PTR) && (flags & CL_MEM_USE_HOST_PTR))
|| ((flags & CL_MEM_COPY_HOST_PTR) && (flags & CL_MEM_USE_HOST_PTR))
|| ((flags & (~(CL_MEM_READ_WRITE | CL_MEM_WRITE_ONLY | CL_MEM_READ_ONLY
| CL_MEM_ALLOC_HOST_PTR | CL_MEM_COPY_HOST_PTR
| CL_MEM_USE_HOST_PTR))) != 0))) {
err = CL_INVALID_VALUE;
goto error;
}
/* This flag is valid only if host_ptr is not NULL */
if (UNLIKELY((((flags & CL_MEM_COPY_HOST_PTR) ||
(flags & CL_MEM_USE_HOST_PTR)) &&
data == NULL))
|| (!(flags & (CL_MEM_COPY_HOST_PTR
|CL_MEM_USE_HOST_PTR))
&& (data != NULL))) {
err = CL_INVALID_HOST_PTR;
goto error;
}
/* CL_MEM_ALLOC_HOST_PTR and CL_MEM_USE_HOST_PTR
are mutually exclusive. */
if (UNLIKELY(flags & CL_MEM_ALLOC_HOST_PTR &&
flags & CL_MEM_USE_HOST_PTR)) {
err = CL_INVALID_HOST_PTR;
goto error;
}
/* CL_MEM_COPY_HOST_PTR and CL_MEM_USE_HOST_PTR
are mutually exclusive. */
if (UNLIKELY(flags & CL_MEM_COPY_HOST_PTR &&
flags & CL_MEM_USE_HOST_PTR)) {
err = CL_INVALID_HOST_PTR;
goto error;
}
if ((err = cl_get_device_info(ctx->device,
CL_DEVICE_MAX_MEM_ALLOC_SIZE,
sizeof(max_mem_size),
&max_mem_size,
NULL)) != CL_SUCCESS) {
goto error;
}
if (UNLIKELY(sz > max_mem_size)) {
err = CL_INVALID_BUFFER_SIZE;
goto error;
}
/* Create the buffer in video memory */
mem = cl_mem_allocate(CL_MEM_BUFFER_TYPE, ctx, flags, sz, CL_FALSE, &err);
if (mem == NULL || err != CL_SUCCESS)
goto error;
/* Copy the data if required */
if (flags & CL_MEM_COPY_HOST_PTR || flags & CL_MEM_USE_HOST_PTR)
cl_buffer_subdata(mem->bo, 0, sz, data);
if (flags & CL_MEM_USE_HOST_PTR || flags & CL_MEM_COPY_HOST_PTR)
mem->host_ptr = data;
exit:
if (errcode_ret)
*errcode_ret = err;
return mem;
error:
cl_mem_delete(mem);
mem = NULL;
goto exit;
}
