avoid global pooling in mkldnn.

zheng-da · zheng-da · commit f6efd342e649 · 2018-02-02T00:00:18.000Z
There is a bug in global pooling in mkldnn.
diff --git 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 &param) {
   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 &param,
