mawei/dragonpilot
1
0
Fork 0
mirror of https://github.com/dragonpilot/dragonpilot.git synced 2026-06-20 21:42:05 +08:00
Code Issues Packages Projects Releases Wiki Activity

Thneed::clinit() : use cl_get_device_id helper(#19934)

Browse Source
This commit is contained in:
Dean Lee
2021-01-27 19:09:28 +08:00
committed by GitHub
parent 064bcbe80b
commit 21f45e3ef4
1 changed files with 4 additions and 19 deletions
Download Patch File Download Diff File Expand all files Collapse all files
selfdrive/modeld/thneed/thneed.cc
+4 -19
View File
@@ -5,6 +5,7 @@
#include <string>
#include <string.h>
#include <errno.h>
#include "common/clutil.h"
#include "thneed.h"
//#define RUN_DISASSEMBLER
@@ -343,26 +344,11 @@ void Thneed::execute(float **finputs, float *foutput, bool slow) {
}
void Thneed::clinit() {
cl_int err;
cl_platform_id platform_id[2];
cl_uint num_devices;
cl_uint num_platforms;
err = clGetPlatformIDs(sizeof(platform_id)/sizeof(cl_platform_id), platform_id, &num_platforms);
assert(err == 0);
err = clGetDeviceIDs(platform_id[0], CL_DEVICE_TYPE_DEFAULT, 1, &device_id, &num_devices);
assert(err == 0);
context = clCreateContext(NULL, 1, &device_id, NULL, NULL, &err);
assert(err == 0);
device_id = cl_get_device_id(CL_DEVICE_TYPE_DEFAULT);
context = CL_CHECK_ERR(clCreateContext(NULL, 1, &device_id, NULL, NULL, &err));
//cl_command_queue_properties props[3] = {CL_QUEUE_PROPERTIES, CL_QUEUE_PROFILING_ENABLE, 0};
cl_command_queue_properties props[3] = {CL_QUEUE_PROPERTIES, 0, 0};
command_queue = clCreateCommandQueueWithProperties(context, device_id, props, &err);
assert(err == 0);
command_queue = CL_CHECK_ERR(clCreateCommandQueueWithProperties(context, device_id, props, &err));
printf("Thneed::clinit done\n");
}
@@ -612,4 +598,3 @@ void CLQueuedKernel::debug_print(bool verbose) {
}
}
}