place.cc 1.7 KB
Newer Older
L
liaogang 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14
/* Copyright (c) 2016 PaddlePaddle Authors. All Rights Reserve.

   Licensed under the Apache License, Version 2.0 (the "License");
   you may not use this file except in compliance with the License.
   You may obtain a copy of the License at

   http://www.apache.org/licenses/LICENSE-2.0

   Unless required by applicable law or agreed to in writing, software
   distributed under the License is distributed on an "AS IS" BASIS,
   WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
   See the License for the specific language governing permissions and
   limitations under the License. */

15
#include "paddle/platform/place.h"
16

17 18
namespace paddle {
namespace platform {
19 20 21

namespace detail {

L
liaogang 已提交
22
class PlacePrinter : public boost::static_visitor<> {
Y
Yi Wang 已提交
23
 public:
L
liaogang 已提交
24
  explicit PlacePrinter(std::ostream &os) : os_(os) {}
25 26
  void operator()(const CPUPlace &) { os_ << "CPUPlace"; }
  void operator()(const GPUPlace &p) { os_ << "GPUPlace(" << p.device << ")"; }
27

Y
Yi Wang 已提交
28
 private:
29
  std::ostream &os_;
30 31
};

Y
Yi Wang 已提交
32
}  // namespace detail
33 34 35

static Place the_default_place;

36 37
void set_place(const Place &place) { the_default_place = place; }
const Place &get_place() { return the_default_place; }
38

39 40
const GPUPlace default_gpu() { return GPUPlace(0); }
const CPUPlace default_cpu() { return CPUPlace(); }
41

42
bool is_gpu_place(const Place &p) {
43
  return boost::apply_visitor(IsGPUPlace(), p);
44
}
45
bool is_cpu_place(const Place &p) {
46
  return !boost::apply_visitor(IsGPUPlace(), p);
47 48
}

49
bool places_are_same_class(const Place &p1, const Place &p2) {
L
liaogang 已提交
50
  return is_gpu_place(p1) == is_gpu_place(p2);
51 52
}

53 54
std::ostream &operator<<(std::ostream &os, const Place &p) {
  detail::PlacePrinter printer(os);
L
liaogang 已提交
55 56
  boost::apply_visitor(printer, p);
  return os;
57 58
}

Y
Yi Wang 已提交
59 60
}  // namespace platform
}  // namespace paddle
反馈
建议
客服 返回
顶部