add example03 for using fp64 extension

This commit is contained in:
Dakota St. Laurent
2015-08-03 06:50:31 -04:00
parent c8179b88a7
commit 3129ddfbf9
2 changed files with 114 additions and 0 deletions

5
.gitignore vendored
View File

@@ -1,5 +1,10 @@
# compiled files
*.out
**/bin
**/build
# clFFT library
clFFT
# THIS IS MY SWAP
# (vim swap files)

109
example03/main.c Normal file
View File

@@ -0,0 +1,109 @@
#include <stdio.h>
#include <stdlib.h>
#include <math.h>
#include <CL/cl.h>
#pragma OPENCL EXTENSION cl_khr_fp64 : enable
const char *kernelSource =
"#pragma OPENCL EXTENSION cl_khr_fp64 : enable \n" \
"__kernel void mult(__global double *v) { \n" \
" int id; \n" \
" id = get_global_id(0); \n" \
" v[id] = 2*v[id]; \n" \
"} \n" \
"\n" ;
int main( int argc, char* argv[] ) {
// problem-related declarations
unsigned int N = 128;
size_t N_bytes = N * sizeof(double);
// openCL declarations
cl_platform_id platform;
cl_device_id device_id;
cl_context context;
cl_command_queue queue;
cl_program program;
cl_kernel k_mult;
// host version of v
double *h_v; // real & imaginary parts
h_v = (double*) malloc(N_bytes);
// initialize v on host
int i;
for (i = 0; i < N; i++) {
h_v[i] = i;
}
// global & local number of threads
size_t globalSize, localSize;
globalSize = N;
localSize = 32;
// show the extensions that are supported
/*
cl_char extensions[2048] = {0};
clGetDeviceInfo(device_id, CL_DEVICE_EXTENSIONS, sizeof(extensions), &extensions, NULL);
printf("%s\n", extensions);
*/
// setup OpenCL stuff
cl_int err;
err = clGetPlatformIDs(1, &platform, NULL);
err = clGetDeviceIDs(platform, CL_DEVICE_TYPE_GPU, 1, &device_id, NULL);
context = clCreateContext(0, 1, &device_id, NULL, NULL, &err);
queue = clCreateCommandQueue(context, device_id, 0, &err);
program = clCreateProgramWithSource(context, 1, (const char **) & kernelSource, NULL, &err);
// Build the program executable
err = clBuildProgram(program, 0, NULL, NULL, NULL, NULL);
if (err != CL_SUCCESS) {
printf("building program failed\n");
if (err == CL_BUILD_PROGRAM_FAILURE) {
size_t log_size;
clGetProgramBuildInfo(program, device_id, CL_PROGRAM_BUILD_LOG, 0, NULL, &log_size);
char *log = (char *) malloc(log_size);
clGetProgramBuildInfo(program, device_id, CL_PROGRAM_BUILD_LOG, log_size, log, NULL);
printf("%s\n", log);
}
}
k_mult = clCreateKernel(program, "mult", &err);
// create arrays on host and write them
cl_mem d_v;
d_v = clCreateBuffer(context, CL_MEM_READ_WRITE, N_bytes, NULL, NULL);
err = clEnqueueWriteBuffer(queue, d_v, CL_TRUE, 0, N_bytes, h_v, 0, NULL, NULL);
err = clSetKernelArg(k_mult, 0, sizeof(cl_mem), &d_v);
err = clEnqueueNDRangeKernel(queue, k_mult, 1, NULL, &globalSize, &localSize, 0, NULL, NULL);
clFinish(queue);
// transfer back
clEnqueueReadBuffer(queue, d_v, CL_TRUE, 0, N_bytes, h_v, 0, NULL, NULL );
clFinish(queue);
int correct = 1;
for (i=0; i<N; i++) {
if (h_v[i] != (double) 2*i)
correct = 0;
}
if (correct)
printf("Array is correct!\n");
else
printf("Array is incorrect :(\n");
// release OpenCL resources
clReleaseMemObject(d_v);
clReleaseProgram(program);
clReleaseKernel(k_mult);
clReleaseCommandQueue(queue);
clReleaseContext(context);
//release host memory
free(h_v);
return 0;
}