62 lines
1.1 KiB
62 lines
1.1 KiB
#include <majel/place.h>
|
|
|
|
namespace majel {
|
|
|
|
namespace detail {
|
|
|
|
class PlacePrinter
|
|
: public boost::static_visitor<> {
|
|
private:
|
|
std::ostream& os_;
|
|
public:
|
|
PlacePrinter(std::ostream& os) : os_(os) {}
|
|
|
|
void operator()(const CpuPlace&) {
|
|
os_ << "CpuPlace";
|
|
}
|
|
|
|
void operator()(const GpuPlace& p) {
|
|
os_ << "GpuPlace(" << p.device << ")";
|
|
}
|
|
};
|
|
|
|
} // namespace majel
|
|
|
|
static Place the_default_place;
|
|
|
|
void set_place(const Place& place) {
|
|
the_default_place = place;
|
|
}
|
|
|
|
const Place& get_place() {
|
|
return the_default_place;
|
|
}
|
|
|
|
const GpuPlace default_gpu() {
|
|
return GpuPlace(0);
|
|
}
|
|
|
|
const CpuPlace default_cpu() {
|
|
return CpuPlace();
|
|
}
|
|
|
|
bool is_gpu_place(const Place& p) {
|
|
return boost::apply_visitor(IsGpuPlace(), p);
|
|
}
|
|
|
|
bool is_cpu_place(const Place& p) {
|
|
return !boost::apply_visitor(IsGpuPlace(), p);
|
|
}
|
|
|
|
bool places_are_same_class(const Place& p1, const Place& p2) {
|
|
return is_gpu_place(p1) == is_gpu_place(p2);
|
|
}
|
|
|
|
std::ostream& operator<<(std::ostream& os, const majel::Place& p) {
|
|
majel::detail::PlacePrinter printer(os);
|
|
boost::apply_visitor(printer, p);
|
|
return os;
|
|
}
|
|
|
|
} // namespace majel
|