提交 05b3196e 编写于 作者: L liaogang

clang-format

上级 d8c8f419
...@@ -4,41 +4,29 @@ namespace majel { ...@@ -4,41 +4,29 @@ namespace majel {
namespace detail { namespace detail {
class PlacePrinter class PlacePrinter : public boost::static_visitor<> {
: public boost::static_visitor<> {
private: private:
std::ostream& os_; std::ostream& os_;
public: public:
PlacePrinter(std::ostream& os) : os_(os) {} PlacePrinter(std::ostream& os) : os_(os) {}
void operator()(const CpuPlace&) { void operator()(const CpuPlace&) { os_ << "CpuPlace"; }
os_ << "CpuPlace";
}
void operator()(const GpuPlace& p) { void operator()(const GpuPlace& p) { os_ << "GpuPlace(" << p.device << ")"; }
os_ << "GpuPlace(" << p.device << ")";
}
}; };
} // namespace majel } // namespace majel
static Place the_default_place; static Place the_default_place;
void set_place(const Place& place) { void set_place(const Place& place) { the_default_place = place; }
the_default_place = place;
}
const Place& get_place() { const Place& get_place() { return the_default_place; }
return the_default_place;
}
const GpuPlace default_gpu() { const GpuPlace default_gpu() { return GpuPlace(0); }
return GpuPlace(0);
}
const CpuPlace default_cpu() { const CpuPlace default_cpu() { return CpuPlace(); }
return CpuPlace();
}
bool is_gpu_place(const Place& p) { bool is_gpu_place(const Place& p) {
return boost::apply_visitor(IsGpuPlace(), p); return boost::apply_visitor(IsGpuPlace(), p);
......
#include "gtest/gtest.h"
#include "majel/place.h" #include "majel/place.h"
#include <sstream> #include <sstream>
#include "gtest/gtest.h"
TEST(Place, Equality) { TEST(Place, Equality) {
majel::CpuPlace cpu; majel::CpuPlace cpu;
...@@ -18,12 +18,12 @@ TEST(Place, Equality) { ...@@ -18,12 +18,12 @@ TEST(Place, Equality) {
} }
TEST(Place, Default) { TEST(Place, Default) {
EXPECT_TRUE(majel::is_gpu_place( majel::get_place())); EXPECT_TRUE(majel::is_gpu_place(majel::get_place()));
EXPECT_TRUE(majel::is_gpu_place( majel::default_gpu())); EXPECT_TRUE(majel::is_gpu_place(majel::default_gpu()));
EXPECT_TRUE(majel::is_cpu_place( majel::default_cpu())); EXPECT_TRUE(majel::is_cpu_place(majel::default_cpu()));
majel::set_place(majel::CpuPlace()); 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) { TEST(Place, Print) {
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册