This is an automated email from the ASF dual-hosted git repository.
kellen pushed a commit to branch v1.4.x
in repository https://gitbox.apache.org/repos/asf/incubator-mxnet.git
The following commit(s) were added to refs/heads/v1.4.x by this push:
new bf86caa fix bipartite match memory corruption (#13900)
bf86caa is described below
commit bf86caa886cfc2e06dde131ed3c290559d418cac
Author: Kellen Sunderland <[email protected]>
AuthorDate: Wed Jan 16 14:35:18 2019 -0800
fix bipartite match memory corruption (#13900)
---
src/operator/contrib/bounding_box-inl.h | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/src/operator/contrib/bounding_box-inl.h
b/src/operator/contrib/bounding_box-inl.h
index 8e96346..031dd95 100644
--- a/src/operator/contrib/bounding_box-inl.h
+++ b/src/operator/contrib/bounding_box-inl.h
@@ -785,7 +785,7 @@ void BipartiteMatchingForward(const nnvm::NodeAttrs& attrs,
.get_with_shape<xpu, 2, DType>(Shape2(batch_size, col), s);
Shape<1> sort_index_shape = Shape1(dshape.Size());
index_t workspace_size = sort_index_shape.Size();
- workspace_size += ((sort_index_shape.Size() * sizeof(int32_t) - 1) /
sizeof(DType)) * 2;
+ workspace_size += (sort_index_shape.Size() * 2 * sizeof(int32_t) - 1) /
sizeof(DType) + 1;
Tensor<xpu, 1, DType> workspace = ctx.requested[0]
.get_space_typed<xpu, 1, DType>(Shape1(workspace_size), s);
Tensor<xpu, 1, DType> scores_copy(workspace.dptr_,
