diff options
| -rw-r--r-- | lib/TH/generic/THTensor.c | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/lib/TH/generic/THTensor.c b/lib/TH/generic/THTensor.c index 38bbffd..2331cdd 100644 --- a/lib/TH/generic/THTensor.c +++ b/lib/TH/generic/THTensor.c @@ -325,6 +325,7 @@ int THTensor_(expand2)(THTensor *ra, THTensor *rb, THTensor *opa, THTensor *opb, opb->size, THTensor_(nDimension)(opb), raiseErrors); if(ret != 0) { + THLongStorage_free(sizes); return ret; } @@ -361,6 +362,9 @@ TH_API int THTensor_(expand3)(THTensor *ra, THTensor *rb, THTensor *rc, THTensor raiseErrors); if(ret != 0) { + THLongStorage_free(sizes); + THFree(op_dims); + THFree(op_sizes); return ret; } @@ -372,6 +376,8 @@ TH_API int THTensor_(expand3)(THTensor *ra, THTensor *rb, THTensor *rc, THTensor THAssert(ret == 0); // since we inferred this already, it must be valid THLongStorage_free(sizes); + THFree(op_dims); + THFree(op_sizes); return 0; } |