cf5028d55b
* Added CUDA support * Performance improvements * API improvements * Added "depth" parameter to GEMM * Android cross-compilation
94 lines
1.7 KiB
C++
94 lines
1.7 KiB
C++
#include "isaac/driver/backend.h"
|
|
#include <assert.h>
|
|
#include <stdexcept>
|
|
|
|
namespace isaac
|
|
{
|
|
|
|
namespace driver
|
|
{
|
|
|
|
queues_type::queues_type(): default_device(0), queue_properties(0)
|
|
{}
|
|
|
|
std::vector<CommandQueue> & queues_type::append(Context const & context)
|
|
{
|
|
data_.push_back(std::make_pair(context, std::vector<CommandQueue>(1, CommandQueue(context, context.device(), queue_properties))));
|
|
return data_.back().second;
|
|
}
|
|
|
|
void queues_type::cuinit()
|
|
{
|
|
#ifdef ISAAC_WITH_CUDA
|
|
cuda::check(cuInit(0));
|
|
int N;
|
|
cuda::check(cuDeviceGetCount(&N));
|
|
for(int i = 0 ; i < N ; ++i)
|
|
append(Context(i));
|
|
#endif
|
|
}
|
|
|
|
void queues_type::clinit()
|
|
{
|
|
std::vector<cl::Platform> platforms;
|
|
cl::Platform::get(&platforms);
|
|
for(auto & p : platforms)
|
|
{
|
|
std::vector<cl::Device> devices;
|
|
p.getDevices(CL_DEVICE_TYPE_ALL, &devices);
|
|
for(auto & d : devices)
|
|
append(Context(d));
|
|
}
|
|
}
|
|
|
|
void queues_type::init()
|
|
{
|
|
if(data_.empty())
|
|
{
|
|
cuinit();
|
|
clinit();
|
|
}
|
|
}
|
|
|
|
std::vector<CommandQueue> & queues_type::operator[](Context const & context)
|
|
{
|
|
init();
|
|
for(auto & x : data_)
|
|
if(x.first==context) return x.second;
|
|
return append(context);
|
|
}
|
|
|
|
Context queues_type::default_context()
|
|
{
|
|
init();
|
|
container_type::iterator it = data_.begin();
|
|
std::advance(it, default_device);
|
|
return it->first;
|
|
}
|
|
|
|
std::vector<CommandQueue> & queues_type::default_queues()
|
|
{ return (*this)[default_context()]; }
|
|
|
|
|
|
queues_type::container_type const & queues_type::contexts()
|
|
{
|
|
init();
|
|
return data_;
|
|
}
|
|
|
|
|
|
void synchronize(std::vector<CommandQueue > & queues)
|
|
{
|
|
for(CommandQueue & q: queues)
|
|
q.synchronize();
|
|
}
|
|
|
|
void synchronize(Context const & context)
|
|
{ synchronize(queues[context]); }
|
|
|
|
queues_type queues;
|
|
|
|
}
|
|
|
|
}
|