clang-format

refactor_docs
liaogang 8 years ago
parent d8c8f419f9
commit 05b3196e0a

@ -4,58 +4,46 @@ namespace majel {
namespace detail {
class PlacePrinter
: public boost::static_visitor<> {
class PlacePrinter : public boost::static_visitor<> {
private:
std::ostream& os_;
std::ostream& os_;
public:
PlacePrinter(std::ostream& os) : os_(os) {}
PlacePrinter(std::ostream& os) : os_(os) {}
void operator()(const CpuPlace&) {
os_ << "CpuPlace";
}
void operator()(const CpuPlace&) { os_ << "CpuPlace"; }
void operator()(const GpuPlace& p) {
os_ << "GpuPlace(" << p.device << ")";
}
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;
}
void set_place(const Place& place) { the_default_place = place; }
const Place& get_place() {
return the_default_place;
}
const Place& get_place() { return the_default_place; }
const GpuPlace default_gpu() {
return GpuPlace(0);
}
const GpuPlace default_gpu() { return GpuPlace(0); }
const CpuPlace default_cpu() {
return CpuPlace();
}
const CpuPlace default_cpu() { return CpuPlace(); }
bool is_gpu_place(const Place& p) {
return boost::apply_visitor(IsGpuPlace(), p);
return boost::apply_visitor(IsGpuPlace(), p);
}
bool is_cpu_place(const Place& p) {
return !boost::apply_visitor(IsGpuPlace(), 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);
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;
majel::detail::PlacePrinter printer(os);
boost::apply_visitor(printer, p);
return os;
}
} // namespace majel

@ -1,6 +1,6 @@
#include "gtest/gtest.h"
#include "majel/place.h"
#include <sstream>
#include "gtest/gtest.h"
TEST(Place, Equality) {
majel::CpuPlace cpu;
@ -18,12 +18,12 @@ TEST(Place, Equality) {
}
TEST(Place, Default) {
EXPECT_TRUE(majel::is_gpu_place( majel::get_place()));
EXPECT_TRUE(majel::is_gpu_place( majel::default_gpu()));
EXPECT_TRUE(majel::is_cpu_place( majel::default_cpu()));
EXPECT_TRUE(majel::is_gpu_place(majel::get_place()));
EXPECT_TRUE(majel::is_gpu_place(majel::default_gpu()));
EXPECT_TRUE(majel::is_cpu_place(majel::default_cpu()));
majel::set_place(majel::CpuPlace());
EXPECT_TRUE(majel::is_cpu_place( majel::get_place()));
EXPECT_TRUE(majel::is_cpu_place(majel::get_place()));
}
TEST(Place, Print) {
@ -33,8 +33,8 @@ TEST(Place, Print) {
EXPECT_EQ("GpuPlace(1)", ss.str());
}
{
std::stringstream ss;
ss << majel::CpuPlace();
EXPECT_EQ("CpuPlace", ss.str());
std::stringstream ss;
ss << majel::CpuPlace();
EXPECT_EQ("CpuPlace", ss.str());
}
}

@ -1,6 +1,6 @@
#include "gtest/gtest.h"
int main(int argc, char** argv) {
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
::testing::InitGoogleTest(&argc, argv);
return RUN_ALL_TESTS();
}

Loading…
Cancel
Save