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

GEMM: Enabled use of cuBLAS when predicted beneficial

Browse Source
This commit is contained in:
Philippe Tillet
2016-10-04 04:20:07 -04:00
parent a4ed0dfbec
commit 3293c45e60
26 changed files with 19391 additions and 26856 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
lib/driver/backend.cpp
View File

@@ -217,9 +217,9 @@ void backend::platforms(std::vector<Platform> & platforms)
if(dispatch::clinit())
{
cl_uint nplatforms;
check(dispatch::dispatch::clGetPlatformIDs(0, NULL, &nplatforms));
dispatch::dispatch::clGetPlatformIDs(0, NULL, &nplatforms);
std::vector<cl_platform_id> clplatforms(nplatforms);
check(dispatch::dispatch::clGetPlatformIDs(nplatforms, clplatforms.data(), NULL));
dispatch::dispatch::clGetPlatformIDs(nplatforms, clplatforms.data(), NULL);
for(cl_platform_id p: clplatforms){
Platform tmp(p);
if(tmp.name().find("CUDA")!=std::string::npos && has_cuda)