diff options
author | Trevor Killeen <killeentm@gmail.com> | 2016-11-17 02:42:06 +0300 |
---|---|---|
committer | Trevor Killeen <killeentm@gmail.com> | 2016-11-17 02:43:57 +0300 |
commit | 45488e235e09f03abb9389508a946b3ef2a6e162 (patch) | |
tree | f34367087626bfc76228a61afa77a4ed52ab6e92 | |
parent | 857f6d218470fb0329a7398a55b39af0a10d6b63 (diff) |
fix memory leak in (equal)
-rw-r--r-- | lib/THC/generic/THCTensorMathPairwise.cu | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/lib/THC/generic/THCTensorMathPairwise.cu b/lib/THC/generic/THCTensorMathPairwise.cu index bfd5464..0b4094b 100644 --- a/lib/THC/generic/THCTensorMathPairwise.cu +++ b/lib/THC/generic/THCTensorMathPairwise.cu @@ -203,6 +203,8 @@ THC_API int THCTensor_(equal)(THCState *state, THCTensor *self_, THCTensor *src_ } unsigned char min = THCudaByteTensor_minall(state, buf); + + THLongStorage_free(size); THCudaByteTensor_free(state, buf); return min != 0; |