Device.hpp

Go to the documentation of this file.

#ifndef CYTNX_DEVICE_H_
#define CYTNX_DEVICE_H_
 
#include <vector>
#include <string>
 
namespace cytnx {
 
  struct __device {
    enum __pybind_device { cpu = -1, cuda = 0 };
  };
 
  class Device_class {
   public:
    enum : int { cpu = -1, cuda = 0 };
    int Ngpus;
    int Ncpus;
    std::vector<std::vector<bool>> CanAccessPeer;
    Device_class();
    void Print_Property();
    std::string getname(const int &device_id);
    ~Device_class();
    // void cudaDeviceSynchronize();
  };
 
  extern Device_class Device;
}  // namespace cytnx
 
#endif  // CYTNX_DEVICE_H_