triton/include/isaac/driver/handle.h

#ifndef ISAAC_DRIVER_HANDLE_H
#define ISAAC_DRIVER_HANDLE_H

#include <map>
#include <memory>

#include "isaac/defines.h"
#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 ISAACAPI 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