From bf269d67b3d1e8d88797022e4ea542f34c06df47 Mon Sep 17 00:00:00 2001 From: tensor-tang Date: Fri, 15 Dec 2017 16:48:00 +0800 Subject: [PATCH] fix place_test on MKLDNNPlace --- paddle/platform/place.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/paddle/platform/place.cc b/paddle/platform/place.cc index 16f17742858..25fe8d21b49 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) { -- GitLab