gaspersic/triton
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

some cleaning

Browse Source
This commit is contained in:
Philippe Tillet
2019-03-21 23:51:47 -07:00
parent 907bbb1ad2
commit 49fd6ece99
20 changed files with 423 additions and 182 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
lib/driver/buffer.cpp
View File

@@ -47,11 +47,11 @@ driver::context* buffer::context() {
}
buffer* buffer::create(driver::context* ctx, size_t size) {
if(dynamic_cast<driver::cu_context*>(ctx))
return new cu_buffer(ctx, size);
if(dynamic_cast<driver::ocl_context*>(ctx))
return new ocl_buffer(ctx, size);
throw std::runtime_error("unknown context");
switch(ctx->backend()){
case CUDA: return new cu_buffer(ctx, size);
case OpenCL: return new ocl_buffer(ctx, size);
default: throw std::runtime_error("unknown backend");
}
}
//
@@ -59,7 +59,8 @@ buffer* buffer::create(driver::context* ctx, size_t size) {
ocl_buffer::ocl_buffer(driver::context* context, size_t size)
: buffer(context, cl_mem(), true){
cl_int err;
dispatch::clCreateBuffer(*context->cl(), CL_MEM_READ_WRITE, size, NULL, &err);
*cl_ = dispatch::clCreateBuffer(*context->cl(), CL_MEM_READ_WRITE, size, NULL, &err);
check(err);
}