place.h

#pragma once
#include <boost/variant.hpp>
#include <iostream>

namespace paddle {
namespace platform {

struct CpuPlace {
  // WORKAROUND: for some reason, omitting this constructor
  // causes errors with boost 1.59 and OSX
  CpuPlace() {}

  // needed for variant equality comparison
  inline bool operator==(const CpuPlace &) const { return true; }
  inline bool operator!=(const CpuPlace &) const { return false; }
};

struct GpuPlace {
  GpuPlace() : GpuPlace(0) {}
  GpuPlace(int d) : device(d) {}

  // needed for variant equality comparison
  inline bool operator==(const GpuPlace &o) const { return device == o.device; }
  inline bool operator!=(const GpuPlace &o) const { return !(*this == o); }

  int device;
};

struct IsGpuPlace : public boost::static_visitor<bool> {
  bool operator()(const CpuPlace &) const { return false; }
  bool operator()(const GpuPlace &gpu) const { return true; }
};

typedef boost::variant<GpuPlace, CpuPlace> Place;

void set_place(const Place &);
const Place &get_place();

const GpuPlace default_gpu();
const CpuPlace default_cpu();

bool is_gpu_place(const Place &);
bool is_cpu_place(const Place &);
bool places_are_same_class(const Place &, const Place &);

std::ostream &operator<<(std::ostream &, const Place &);

}  // namespace platform
}  // namespace paddle