avoid global pooling in mkldnn.

There is a bug in global pooling in mkldnn.
zheng-da · Feb 2, 2018 · f6efd34 · f6efd34
1 parent b48346a
commit f6efd34
Showing 1 changed file with 3 additions and 1 deletion.
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,