Skip to content

Commit

Permalink
src: add error handling and print platform info
Browse files Browse the repository at this point in the history 
Signed-off-by: Daniel Bevenius <[email protected]>
  • Loading branch information
@danbev
danbev committed Jun 1, 2024
1 parent 3f8d524 commit 79c46ec
Showing 1 changed file with 110 additions and 23 deletions.
133 changes: 110 additions & 23 deletions gpu/opencl/src/main.c
View file
Original file line number Diff line number Diff line change
@@ -1,11 +1,38 @@
// main.c
#include <stdio.h>
#include <stdlib.h>

#define CL_TARGET_OPENCL_VERSION 300
#include <CL/cl.h>

#define NUM_ELEMENTS 1024

const char *kernelSource =
"__kernel void add_vectors(__global const float *a, __global const float *b, __global float *result, unsigned int numElements) {\n"
" int id = get_global_id(0);\n"
" if (id < numElements) {\n"
" result[id] = a[id] + b[id];\n"
" }\n"
"}\n";

void printPlatformInfo(cl_platform_id platform) {
char info[1024];
size_t infoSize;

clGetPlatformInfo(platform, CL_PLATFORM_NAME, sizeof(info), info, &infoSize);
printf("\n-------------------------------\n");
printf("Platform Name: %s\n", info);

clGetPlatformInfo(platform, CL_PLATFORM_VENDOR, sizeof(info), info, &infoSize);
printf("Platform Vendor: %s\n", info);

clGetPlatformInfo(platform, CL_PLATFORM_VERSION, sizeof(info), info, &infoSize);
printf("Platform Version: %s\n", info);

clGetPlatformInfo(platform, CL_PLATFORM_PROFILE, sizeof(info), info, &infoSize);
printf("Platform Profile: %s\n", info);
printf("-------------------------------\n");
}

int main() {
printf("OpenCL Exploration...!\n");

Expand All @@ -18,29 +45,63 @@ int main() {
printf("Error: Failed to get the number of platforms: cl_err: %d\n", err);
return 1;
}
printf("Number of platforms: %d\n", numPlatforms);
cl_platform_id *platforms = (cl_platform_id *)malloc(sizeof(cl_platform_id) * numPlatforms);
err = clGetPlatformIDs(numPlatforms, platforms, NULL);
if (err != CL_SUCCESS) {
printf("Error: Failed to get the number array or cl_platform_id: cl_err: %d\n", err);
return 1;
}


cl_platform_id platform;
clGetPlatformIDs(1, &platform, NULL);
for (int i = 0; i < numPlatforms; i++) {
printPlatformInfo(platforms[i]);
}

cl_device_id device;
clGetDeviceIDs(platform, CL_DEVICE_TYPE_GPU, 1, &device, NULL);
err = clGetDeviceIDs(platforms[1], CL_DEVICE_TYPE_GPU, 1, &device, NULL);
if (err != CL_SUCCESS) {
printf("Error: Failed to get device id: cl_err: %d\n", err);
return 1;
}

cl_context context = clCreateContext(NULL, 1, &device, NULL, NULL, &err);
if (err != CL_SUCCESS) {
printf("Error: Failed to create context: cl_err: %d\n", err);
return 1;
}

cl_context context = clCreateContext(NULL, 1, &device, NULL, NULL, NULL);
cl_queue_properties properties[] = { CL_QUEUE_PROPERTIES, 0, 0 };
cl_command_queue queue = clCreateCommandQueueWithProperties(context, device, properties, NULL);
cl_command_queue queue = clCreateCommandQueueWithProperties(context, device, properties, &err);
if (err != CL_SUCCESS) {
printf("Error: Failed to create command queue: cl_err: %d\n", err);
return 1;
}

unsigned int numElements = NUM_ELEMENTS;

cl_program program = clCreateProgramWithSource(context, 1, &kernelSource, NULL, &err);
if (err != CL_SUCCESS) {
printf("Error: Failed to create program: cl_err: %d\n", err);
return 1;
}

const char *source =
"#pragma OPENCL EXTENSION cl_khr_fp64 : enable\n"
"#include \"add_vectors.cl\"\n";

cl_program program = clCreateProgramWithSource(context, 1, &source, NULL, NULL);
clBuildProgram(program, 1, &device, NULL, NULL, NULL);
err = clBuildProgram(program, 1, &device, NULL, NULL, NULL);
if (err != CL_SUCCESS) {
// Print the build log if there's an error
size_t log_size;
clGetProgramBuildInfo(program, device, CL_PROGRAM_BUILD_LOG, 0, NULL, &log_size);
char *log = (char *) malloc(log_size);
clGetProgramBuildInfo(program, device, CL_PROGRAM_BUILD_LOG, log_size, log, NULL);
printf("Build error:\n%s\n", log);
free(log);
return 1;
}

cl_kernel kernel = clCreateKernel(program, "add_vectors", NULL);
cl_kernel kernel = clCreateKernel(program, "add_vectors", &err);
if (err != CL_SUCCESS) {
printf("Error: Failed to create kernel: cl_err: %d\n", err);
return 1;
}

float *a = (float *)malloc(NUM_ELEMENTS * sizeof(float));
float *b = (float *)malloc(NUM_ELEMENTS * sizeof(float));
Expand All @@ -52,22 +113,48 @@ int main() {
}

cl_mem bufferA = clCreateBuffer(context, CL_MEM_READ_ONLY | CL_MEM_COPY_HOST_PTR,
NUM_ELEMENTS * sizeof(float), a, NULL);
NUM_ELEMENTS * sizeof(float), a, &err);
if (err != CL_SUCCESS) {
printf("Error: Failed to create bufferA: cl_err: %d\n", err);
return 1;
}
cl_mem bufferB = clCreateBuffer(context, CL_MEM_READ_ONLY | CL_MEM_COPY_HOST_PTR,
NUM_ELEMENTS * sizeof(float), b, NULL);
NUM_ELEMENTS * sizeof(float), b, &err);
if (err != CL_SUCCESS) {
printf("Error: Failed to create bufferB: cl_err: %d\n", err);
return 1;
}
cl_mem bufferResult = clCreateBuffer(context, CL_MEM_WRITE_ONLY,
NUM_ELEMENTS * sizeof(float), NULL, NULL);
NUM_ELEMENTS * sizeof(float), NULL, &err);
if (err != CL_SUCCESS) {
printf("Error: Failed to create bufferResult: cl_err: %d\n", err);
return 1;
}

clSetKernelArg(kernel, 0, sizeof(cl_mem), &bufferA);
clSetKernelArg(kernel, 1, sizeof(cl_mem), &bufferB);
clSetKernelArg(kernel, 2, sizeof(cl_mem), &bufferResult);
clSetKernelArg(kernel, 3, sizeof(unsigned int), &numElements);
err = clSetKernelArg(kernel, 0, sizeof(cl_mem), &bufferA);
err |= clSetKernelArg(kernel, 1, sizeof(cl_mem), &bufferB);
err |= clSetKernelArg(kernel, 2, sizeof(cl_mem), &bufferResult);
err |= clSetKernelArg(kernel, 3, sizeof(unsigned int), &numElements);
if (err != CL_SUCCESS) {
printf("Error: Failed to set kernel arguments: cl_err: %d\n", err);
return 1;
}

size_t globalSize = NUM_ELEMENTS;
clEnqueueNDRangeKernel(queue, kernel, 1, NULL, &globalSize, NULL, 0, NULL, NULL);
err = clEnqueueNDRangeKernel(queue, kernel, 1, NULL, &globalSize, NULL, 0, NULL, NULL);
if (err != CL_SUCCESS) {
printf("Error: Failed to enqueue kernel: cl_err: %d\n", err);
return 1;
}


clEnqueueReadBuffer(queue, bufferResult, CL_TRUE, 0,
err = clEnqueueReadBuffer(queue, bufferResult, CL_TRUE, 0,
NUM_ELEMENTS * sizeof(float), result, 0, NULL, NULL);
if (err != CL_SUCCESS) {
printf("Error: Failed to read buffer: cl_err: %d\n", err);
return 1;
}


for (int i = 0; i < 10; i++) {
printf("%.2f + %.2f = %.2f\n", a[i], b[i], result[i]);
Expand Down

0 comments on commit 79c46ec

Please sign in to comment.