fix_uninitialized_issue

apache · Mar 17, 2020 · 009e538 · 009e538
1 parent 97dd362
commit 009e538
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/src/operator/numpy/np_interp_op-inl.h b/src/operator/numpy/np_interp_op-inl.h
@@ -228,13 +228,15 @@ void NumpyInterpForward(const nnvm::NodeAttrs& attrs,
 
   TBlob x, idx;
   if (x_scalar.has_value()) {
-    double x_value = x_scalar.value();
-    Tensor<cpu, 1, double> host_x(&x_value, Shape1(1), ctx.get_stream<cpu>());
+    double* x_value_ptr = new double;
+    *x_value_ptr = x_scalar.value();
+    Tensor<cpu, 1, double> host_x(x_value_ptr, Shape1(1), ctx.get_stream<cpu>());
     Tensor<xpu, 1, double> device_x(reinterpret_cast<double*>(workspace_curr_ptr),
                                     Shape1(1), ctx.get_stream<xpu>());
     Copy(device_x, host_x, ctx.get_stream<xpu>());
     x = TBlob(device_x.dptr_, TShape(0, 1), xpu::kDevMask);
     workspace_curr_ptr += 8;
+    delete x_value_ptr;
   } else {
     x = inputs[2];
   }   // handle input x is a scalar