fix place_test on MKLDNNPlace

8 years ago · bf269d67b3
parent a92f057ed1
commit bf269d67b3
1 changed files with 1 additions and 1 deletions
--- a/paddle/platform/place.cc
+++ b/paddle/platform/place.cc
@ -45,7 +45,7 @@ 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);
+  return !is_gpu_place(p) && !is_mkldnn_place(p);
 }

 bool is_mkldnn_place(const Place &p) {