1 | #include "NBody.hpp" |
---|
2 | |
---|
3 | /////////////////////////////////// NBody::setupCLProgram() Func \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ |
---|
4 | // |
---|
5 | // Build CL kernel program executable. |
---|
6 | // |
---|
7 | /////////////////////////////////// NBody::setupCLProgram() Func \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ |
---|
8 | |
---|
9 | int NBody::setupCLProgram() |
---|
10 | { |
---|
11 | cl_int status; |
---|
12 | |
---|
13 | /* create a CL program using the kernel source */ |
---|
14 | char *program_source = load_program_source(kernelFileName); |
---|
15 | size_t sourceSize[] = { strlen(program_source) }; |
---|
16 | |
---|
17 | program = clCreateProgramWithSource( |
---|
18 | context, |
---|
19 | 1, |
---|
20 | (const char**)&program_source, |
---|
21 | sourceSize, |
---|
22 | &status); |
---|
23 | |
---|
24 | if (status != CL_SUCCESS){ |
---|
25 | std::cout << "clCreateProgramWithSource failed." << std::endl; |
---|
26 | return 1; |
---|
27 | } |
---|
28 | |
---|
29 | /* create a cl program executable for all the devices specified */ |
---|
30 | status = clBuildProgram( |
---|
31 | program, |
---|
32 | 1, |
---|
33 | &devices[0], |
---|
34 | NULL, |
---|
35 | NULL, |
---|
36 | NULL); |
---|
37 | |
---|
38 | if(status != CL_SUCCESS) |
---|
39 | { |
---|
40 | if(status == CL_BUILD_PROGRAM_FAILURE) |
---|
41 | { |
---|
42 | cl_int logStatus; |
---|
43 | char * buildLog = NULL; |
---|
44 | size_t buildLogSize = 0; |
---|
45 | logStatus = clGetProgramBuildInfo (program, |
---|
46 | devices[0], |
---|
47 | CL_PROGRAM_BUILD_LOG, |
---|
48 | buildLogSize, |
---|
49 | buildLog, |
---|
50 | &buildLogSize); |
---|
51 | if (logStatus != CL_SUCCESS){ |
---|
52 | std::cout << "clGetProgramBuildInfo failed." << std::endl; |
---|
53 | return 1; |
---|
54 | } |
---|
55 | |
---|
56 | buildLog = (char*)malloc(buildLogSize); |
---|
57 | if(buildLog == NULL) |
---|
58 | { |
---|
59 | std::cout << "Failed to allocate host memory. (buildLog)" << std::endl; |
---|
60 | return 1; |
---|
61 | } |
---|
62 | memset(buildLog, 0, buildLogSize); |
---|
63 | |
---|
64 | logStatus = clGetProgramBuildInfo (program, |
---|
65 | devices[0], |
---|
66 | CL_PROGRAM_BUILD_LOG, |
---|
67 | buildLogSize, |
---|
68 | buildLog, |
---|
69 | NULL); |
---|
70 | if (logStatus != CL_SUCCESS){ |
---|
71 | std::cout << "clGetProgramBuildInfo failed." << std::endl; |
---|
72 | free(buildLog); |
---|
73 | return 1; |
---|
74 | } |
---|
75 | |
---|
76 | |
---|
77 | std::cout << " \n\t\t\tBUILD LOG\n"; |
---|
78 | std::cout << " ************************************************\n"; |
---|
79 | std::cout << buildLog << std::endl; |
---|
80 | std::cout << " ************************************************\n"; |
---|
81 | free(buildLog); |
---|
82 | } |
---|
83 | |
---|
84 | if(status != CL_SUCCESS) |
---|
85 | { |
---|
86 | std::cout << "clBuildProgram failed." << std::endl; |
---|
87 | return 1; |
---|
88 | } |
---|
89 | } |
---|
90 | } |
---|
91 | |
---|
92 | /////////////////////////////////// NBody::load_program_source Func \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ |
---|
93 | // |
---|
94 | // Loads the kernel code form a .cl file type and returns the source code as a char* |
---|
95 | // |
---|
96 | /////////////////////////////////// NBody::load_program_source Func \\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\\ |
---|
97 | |
---|
98 | char* NBody::load_program_source(const char *filename) |
---|
99 | { |
---|
100 | FILE *fh; |
---|
101 | char *source; |
---|
102 | int size; |
---|
103 | |
---|
104 | fh = fopen(filename, "r"); |
---|
105 | if (fh == NULL){ |
---|
106 | std::cout << "Reading the source cod for kernel failed." << std::endl; |
---|
107 | exit(1); |
---|
108 | } |
---|
109 | |
---|
110 | fseek (fh, 0, SEEK_END); |
---|
111 | size = ftell (fh); |
---|
112 | rewind(fh); |
---|
113 | |
---|
114 | source = (char *) malloc(size+1); |
---|
115 | |
---|
116 | int result = fread(source, 1, size, fh); |
---|
117 | /*if (result != size){ |
---|
118 | std::cout << "Reading the source cod for kernel failed." << std::endl; |
---|
119 | exit(1); |
---|
120 | }*/ |
---|
121 | source[result] = '\0'; |
---|
122 | |
---|
123 | fclose(fh); |
---|
124 | |
---|
125 | return source; |
---|
126 | } |
---|