From f6efd342e64968cb848c9193d80e929968b8052c Mon Sep 17 00:00:00 2001 From: Da Zheng Date: Mon, 29 Jan 2018 19:41:16 +0000 Subject: [PATCH] avoid global pooling in mkldnn. There is a bug in global pooling in mkldnn. --- src/operator/nn/mkldnn/mkldnn_pooling-inl.h | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/src/operator/nn/mkldnn/mkldnn_pooling-inl.h b/src/operator/nn/mkldnn/mkldnn_pooling-inl.h index 61895b4d4423..4f2f71866e14 100644 --- a/src/operator/nn/mkldnn/mkldnn_pooling-inl.h +++ b/src/operator/nn/mkldnn/mkldnn_pooling-inl.h @@ -81,7 +81,9 @@ class MKLDNNPoolingFwd { inline bool SupportMKLDNNPooling(const PoolingParam ¶m) { return param.kernel.ndim() == 2 && (param.pool_type == pool_enum::kMaxPooling || - param.pool_type == pool_enum::kAvgPooling); + param.pool_type == pool_enum::kAvgPooling) + // This is a temporary fix. There is a bug in global pooling of MKLDNN. + && !param.global_pool; } inline bool SupportMKLDNNPooling(const PoolingParam ¶m,