From 3e7ad44239438cb6132a6c22663596407b38a58a Mon Sep 17 00:00:00 2001 From: Kellen Sunderland Date: Fri, 21 Sep 2018 09:01:50 -0700 Subject: [PATCH] [MXNET-953] Fix oob memory read --- src/operator/tensor/elemwise_unary_op_basic.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/operator/tensor/elemwise_unary_op_basic.cc b/src/operator/tensor/elemwise_unary_op_basic.cc index c3e9c2dc91d0..3da9ff413294 100644 --- 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)); }