fix cpp test failure

apache · May 15, 2019 · f0059f6 · f0059f6
1 parent 812c57f
commit f0059f6
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/operator/nn/mkldnn/mkldnn_sum.cc b/src/operator/nn/mkldnn/mkldnn_sum.cc
@@ -141,18 +141,19 @@ void MKLDNNSumForward(const nnvm::NodeAttrs& attrs, const OpContext &ctx,
       in_bufs[i] = inputs[i].Reorder2Default();
       in_mem = in_bufs[i].GetMKLDNNData();
     } else {
+      in_bufs[i] = inputs[i];
       in_mem = inputs[i].GetMKLDNNData();
     }
     mkldnn::memory::primitive_desc tmp_pd = in_mem->get_primitive_desc();
     data_md.push_back(tmp_pd);
     data_mem.push_back(in_mem);
   }
 
-  MKLDNNSumFwd &fwd = GetSumForward(scales, inputs, data_md);
+  MKLDNNSumFwd &fwd = GetSumForward(scales, in_bufs, data_md);
   mxnet::mkldnn_output_t out_mem = CreateMKLDNNMem(out_data,
                                                    fwd.fwd_pd.dst_primitive_desc(),
                                                    req,
-                                                   &inputs[0]);
+                                                   &in_bufs[0]);
   fwd.SetNewMem(data_mem, *out_mem.second);
   MKLDNNStream::Get()->RegisterPrim(fwd.GetFwd());
   CommitOutput(out_data, out_mem);