diff --git a/paddle/platform/place.cc b/paddle/platform/place.cc index 16f177428589f0fdc170dd1614d94c4abafb64d2..25fe8d21b49b07a6afe2938245906dc1bdd90398 100644 --- 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) {