diff options
| author | Adam Paszke <adam.paszke@gmail.com> | 2016-12-16 00:17:37 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2016-12-16 00:17:37 +0300 |
| commit | 9a338fa4a61c7622aff75ee4d92adddd2accebf8 (patch) | |
| tree | 66e3a74823941a93c9ae4ad92a01cd9dc0e5ceb0 | |
| parent | 59f0492b81e43b5b06a926d651eca5434cf8f259 (diff) | |
Add missing free in LookupTable (#400)
| -rw-r--r-- | lib/THCUNN/generic/LookupTable.cu | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/lib/THCUNN/generic/LookupTable.cu b/lib/THCUNN/generic/LookupTable.cu index 85423e1..d467c6a 100644 --- a/lib/THCUNN/generic/LookupTable.cu +++ b/lib/THCUNN/generic/LookupTable.cu @@ -42,6 +42,7 @@ void THNN_(LookupTable_accGradParameters)( numel, stride, paddingValue); + THCTensor_(free)(state, gradOutput); THCudaCheck(cudaGetLastError()); return; } |