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

No more random order in default initialization

Browse Source 
Added linking to MKL/CBLAS
This commit is contained in:
Philippe Tillet
2015-01-24 14:51:48 -05:00
parent 9a76be3edc
commit da9defac04
12 changed files with 122 additions and 75 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
lib/cl/queues.cpp
View File

@@ -1,5 +1,6 @@
#include "atidlas/cl/queues.h"
#include <assert.h>
#include <stdexcept>
namespace atidlas
{
@@ -9,7 +10,7 @@ namespace cl
void synchronize(cl::Context const & context)
{
std::vector<cl::CommandQueue> & q = queues[context];
std::vector<cl::CommandQueue> & q = get_queues(context);
for(std::vector<cl::CommandQueue>::iterator it = q.begin() ; it != q.end() ; ++it)
it->finish();
}
@@ -30,15 +31,32 @@ queues_t init_queues()
std::vector<cl::Device> current(1, *itt);
cl::Context context(current);
cl::CommandQueue queue(context, *itt);
result.insert(std::make_pair(context, std::vector<cl::CommandQueue>(1, queue)));
result.push_back(std::make_pair(context, std::vector<cl::CommandQueue>(1, queue)));
}
}
return result;
}
cl::Context default_context()
{ return queues.begin()->second.front().getInfo<CL_QUEUE_CONTEXT>(); }
{
return queues[default_context_idx].second.front().getInfo<CL_QUEUE_CONTEXT>();
}
std::vector<cl::CommandQueue> & get_queues(cl::Context const & ctx)
{
for(queues_t::iterator it = queues.begin() ; it != queues.end() ; ++it)
if(it->first()==ctx())
return it->second;
throw std::out_of_range("The context provided is not registered");
}
cl::CommandQueue & get_queue(cl::Context const & ctx, std::size_t idx)
{ return get_queues(ctx)[idx]; }
unsigned int default_context_idx = 0;
queues_t queues = init_queues();
kernels_t kernels = kernels_t();