You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
51 lines
1.3 KiB
51 lines
1.3 KiB
8 years ago
|
#pragma once
|
||
|
|
#include <boost/variant.hpp>
|
||
|
|
#include <iostream>
|
||
|
|
|
||
|
|
namespace majel {
|
||
|
|
|
||
|
|
struct CpuPlace {
|
||
|
|
CpuPlace() {} // WORKAROUND: for some reason, omitting this constructor
|
||
|
|
// causes errors with boost 1.59 and OSX
|
||
|
|
// needed for variant equality comparison
|
||
|
|
inline bool operator==(const CpuPlace&) const { return true; }
|
||
|
|
|
||
|
|
inline bool operator!=(const CpuPlace&) const { return false; }
|
||
|
|
};
|
||
|
|
|
||
|
|
struct GpuPlace {
|
||
|
|
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); }
|
||
|
|
|
||
|
|
GpuPlace() : GpuPlace(0) {}
|
||
|
|
int device;
|
||
|
|
};
|
||
|
|
|
||
|
|
class IsGpuPlace : public boost::static_visitor<bool> {
|
||
|
|
public:
|
||
|
|
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 majel::Place&);
|
||
|
|
|
||
|
|
} // namespace majel
|