use data_type()

apache · Mar 25, 2020 · dc255c7 · dc255c7
1 parent 32f8700
commit dc255c7
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/operator/nn/mkldnn/mkldnn_pooling.cc b/src/operator/nn/mkldnn/mkldnn_pooling.cc
@@ -30,9 +30,8 @@
 namespace mxnet {
 namespace op {
 
-// This can be replaced by mkldnn::memory::desc::data_type() after 1.2 release
 static inline mkldnn::memory::data_type get_data_type(const mkldnn::memory::desc &md) {
-  return static_cast<mkldnn::memory::data_type>(md.data.data_type);
+  return static_cast<mkldnn::memory::data_type>(md.data_type());
 }
 
 void MKLDNNPoolingFwd::Init(const mxnet::NDArray &input, const mxnet::NDArray &output,