use reorder member fn

apache · Jun 12, 2018 · 9a4b299 · 9a4b299
1 parent c0b2179
commit 9a4b299
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/src/operator/nn/mkldnn/mkldnn_sum.cc b/src/operator/nn/mkldnn/mkldnn_sum.cc
@@ -72,10 +72,7 @@ void MKLDNNSumForward(const nnvm::NodeAttrs& attrs, const OpContext &ctx,
       in_mem = inputs[i].GetMKLDNNData(target_pd);
     }
     if (in_mem == nullptr) {
-      auto tmp_memory = TmpMemMgr::Get()->Alloc(target_pd);
-      auto input_memory = inputs[i].GetMKLDNNData();
-      CopyMKLDNNMem(*input_memory, tmp_memory);
-      in_mem = tmp_memory;
+      in_mem = inputs[i].GetMKLDNNDataReorder(target_pd);
     }
     in_prims.push_back(*in_mem);
     in_pds[i] = in_mem->get_primitive_desc();