fix place_test on MKLDNNPlace

bf269d67 · tensor-tang · a92f057e · bf269d67
显示空白变更内容
内联并排

Showing with 1 addition and 1 deletion

paddle/platform/place.cc paddle/platform/place.cc +1 -1

未找到文件。
--- 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) {