[MXNET-953] Fix oob memory read

apache · Sep 21, 2018 · 3e7ad44 · 3e7ad44
1 parent 1045140
commit 3e7ad44
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/operator/tensor/elemwise_unary_op_basic.cc b/src/operator/tensor/elemwise_unary_op_basic.cc
@@ -542,7 +542,7 @@ void SizeComputeCPU(const nnvm::NodeAttrs& attrs,
   CHECK_EQ(req.size(), 1U);
   const TBlob& in_data = inputs[0];
   const TBlob& out_data = outputs[0];
-  const index_t size_var = in_data.Size();
+  const int64_t size_var = in_data.Size();
   memcpy(out_data.dptr_, &size_var, 1U * sizeof(int64_t));
 }