opencl-template/main.c

125 lines
2.9 KiB
C
Raw Normal View History

2018-04-23 09:09:14 +02:00
#include <stdio.h>
#include <CL/cl.h>
#include <stdlib.h>
#include <stdint.h>
2018-04-23 12:56:24 +02:00
#include <string.h>
2018-04-23 09:09:14 +02:00
2018-04-23 12:56:24 +02:00
void check_error(cl_int error)
{
if (error != CL_SUCCESS) {
printf("OpenCL call failed with error %d\n", (int)error);
exit(error);
}
}
void load_kernel_from_file(char *file, char **src)
{
long fsize;
FILE *f = fopen(file, "rb");
fseek(f, 0, SEEK_END);
fsize = ftell(f);
fseek(f, 0, SEEK_SET); //same as rewind(f);
*src = (char *)malloc(fsize + 1);
fread(*src, fsize, 1, f);
fclose(f);
(*src)[fsize] = 0;
}
cl_program create_program(char *src, cl_context context)
{
cl_program program;
size_t lengths[1];
cl_int error = 0;
lengths[0] = strlen(src);
printf("strlen: %u\n", (unsigned int)lengths[0]);
program = clCreateProgramWithSource(context, 1, (const char **)&src,
lengths, &error);
check_error(error);
return program;
}
float test_a[] = {21.321, 213.213, 3213.23};
float test_b[] = {21.21, 13.213, 17.2553};
int main(void)
2018-04-23 09:09:14 +02:00
{
cl_int error;
cl_platform_id *platform_ids = NULL;
cl_device_id *device_ids = NULL;
cl_uint platform_id_count = 0;
cl_uint device_id_count = 0;
cl_context_properties context_properties[4];
cl_context context;
2018-04-23 12:56:24 +02:00
cl_program program;
cl_kernel kernel;
cl_mem a_buff, b_buff;
char *source_code = NULL;
2018-04-23 09:09:14 +02:00
2018-04-23 12:56:24 +02:00
clGetPlatformIDs(0, NULL, &platform_id_count);
2018-04-23 09:09:14 +02:00
if (platform_id_count == 0)
return -1;
2018-04-23 12:56:24 +02:00
platform_ids = (cl_platform_id *)malloc(platform_id_count *
2018-04-23 09:09:14 +02:00
sizeof(cl_platform_id));
if (platform_ids == NULL)
return -1;
2018-04-23 12:56:24 +02:00
clGetPlatformIDs(platform_id_count, platform_ids, NULL);
2018-04-23 09:09:14 +02:00
printf("Platforms available: %u\n", (unsigned int)platform_id_count);
2018-04-23 12:56:24 +02:00
clGetDeviceIDs(platform_ids[0], CL_DEVICE_TYPE_ALL, 0, NULL,
2018-04-23 09:09:14 +02:00
&device_id_count);
2018-04-23 12:56:24 +02:00
printf("Device in Platform 0 count: %u\n",
(unsigned int)device_id_count);
2018-04-23 09:09:14 +02:00
if (device_id_count == 0)
return -1;
2018-04-23 12:56:24 +02:00
device_ids = (cl_device_id *) malloc(sizeof(cl_device_id)
*device_id_count);
2018-04-23 09:09:14 +02:00
2018-04-23 12:56:24 +02:00
clGetDeviceIDs(platform_ids[0], CL_DEVICE_TYPE_ALL, device_id_count,
2018-04-23 09:09:14 +02:00
device_ids, NULL);
context_properties[0] = CL_CONTEXT_PLATFORM;
context_properties[1] = (cl_context_properties)platform_ids[0];
context_properties[2] = context_properties[3] = 0;
2018-04-23 12:56:24 +02:00
context = clCreateContext(context_properties, device_id_count,
2018-04-23 09:09:14 +02:00
device_ids, NULL,
NULL, &error);
2018-04-23 12:56:24 +02:00
check_error(error);
load_kernel_from_file("cl_kernels/saxpy.cl", &source_code);
program = create_program(source_code, context);
/* Compile the source code */
check_error(clBuildProgram(program, device_id_count,
device_ids, NULL, NULL, NULL));
kernel = clCreateKernel(program, "SAXPY", &error);
/* Create memory buffers for data */
a_buff = clCreateBuffer(context, CL_MEM_READ_ONLY |
CL_MEM_COPY_HOST_PTR,
sizeof(test_a), test_a, &error);
check_error(error);
b_buff = clCreateBuffer(context, CL_MEM_READ_WRITE |
CL_MEM_COPY_HOST_PTR,
sizeof(test_b), test_b, &error);
check_error(error);
2018-04-23 09:09:14 +02:00
return 0;
}