cf5028d55b
* Added CUDA support * Performance improvements * API improvements * Added "depth" parameter to GEMM * Android cross-compilation
53 lines
997 B
C++
53 lines
997 B
C++
#ifndef ISAAC_DRIVER_HANDLE_H
|
|
#define ISAAC_DRIVER_HANDLE_H
|
|
|
|
#include <map>
|
|
#include <memory>
|
|
|
|
#include "isaac/driver/common.h"
|
|
#include <iostream>
|
|
namespace isaac
|
|
{
|
|
|
|
namespace driver
|
|
{
|
|
|
|
#ifdef ISAAC_WITH_CUDA
|
|
#define HANDLE_TYPE(CLTYPE, CUTYPE) Handle<CLTYPE, CUTYPE>
|
|
#else
|
|
#define HANDLE_TYPE(CLTYPE, CUTYPE) Handle<CLTYPE, void>
|
|
#endif
|
|
|
|
template<class CLType, class CUType>
|
|
class Handle
|
|
{
|
|
private:
|
|
#ifdef ISAAC_WITH_CUDA
|
|
static void _delete(CUcontext x);
|
|
static void _delete(CUdeviceptr x);
|
|
static void _delete(CUstream x);
|
|
static void _delete(CUdevice);
|
|
static void _delete(CUevent x);
|
|
static void _delete(CUfunction);
|
|
static void _delete(CUmodule x);
|
|
static void _delete(std::pair<CUevent, CUevent> x);
|
|
#endif
|
|
|
|
public:
|
|
Handle(backend_type backend);
|
|
bool operator==(Handle const & other) const;
|
|
bool operator<(Handle const & other) const;
|
|
~Handle();
|
|
|
|
public:
|
|
std::shared_ptr<CLType> cl;
|
|
std::shared_ptr<CUType> cu;
|
|
private:
|
|
backend_type backend_;
|
|
};
|
|
|
|
}
|
|
}
|
|
|
|
#endif
|