@@ -31,14 +31,6 @@ struct CPUPlace {
3131 inline bool operator !=(const CPUPlace &) const { return false ; }
3232};
3333
34- struct MKLDNNPlace {
35- MKLDNNPlace () {}
36-
37- // needed for variant equality comparison
38- inline bool operator ==(const MKLDNNPlace &) const { return true ; }
39- inline bool operator !=(const MKLDNNPlace &) const { return false ; }
40- };
41-
4234struct CUDAPlace {
4335 CUDAPlace () : CUDAPlace(0 ) {}
4436 explicit CUDAPlace (int d) : device(d) {}
@@ -53,37 +45,21 @@ struct CUDAPlace {
5345 int device;
5446};
5547
56- struct CUDNNPlace : public CUDAPlace {
57- CUDNNPlace () : CUDAPlace() {}
58- explicit CUDNNPlace (int d) : CUDAPlace(d) {}
59- };
60-
6148struct IsCUDAPlace : public boost ::static_visitor<bool > {
6249 bool operator ()(const CPUPlace &) const { return false ; }
63- bool operator ()(const MKLDNNPlace &) const { return false ; }
6450 bool operator ()(const CUDAPlace &gpu) const { return true ; }
65- bool operator ()(const CUDNNPlace &) const { return true ; }
66- };
67-
68- struct IsMKLDNNPlace : public boost ::static_visitor<bool > {
69- bool operator ()(const MKLDNNPlace &) const { return true ; }
70- bool operator ()(const CPUPlace &) const { return false ; }
71- bool operator ()(const CUDAPlace &) const { return false ; }
72- bool operator ()(const CUDNNPlace &) const { return false ; }
7351};
7452
75- typedef boost::variant<CUDNNPlace, CUDAPlace, CPUPlace, MKLDNNPlace > Place;
53+ typedef boost::variant<CUDAPlace, CPUPlace> Place;
7654
7755void set_place (const Place &);
7856const Place &get_place ();
7957
8058const CUDAPlace default_gpu ();
8159const CPUPlace default_cpu ();
82- const MKLDNNPlace default_mkldnn ();
8360
8461bool is_gpu_place (const Place &);
8562bool is_cpu_place (const Place &);
86- bool is_mkldnn_place (const Place &);
8763bool places_are_same_class (const Place &, const Place &);
8864
8965std::ostream &operator <<(std::ostream &, const Place &);
0 commit comments