diff options
| author | Soumith Chintala <soumith@gmail.com> | 2015-10-02 18:20:49 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2015-10-02 18:20:49 +0300 |
| commit | b0eeda8bf076587f70d7db953f0f37eb1d5e53d7 (patch) | |
| tree | d62af25c4f93c8e2f4740808df38138c08dfd34d | |
| parent | c34a4a5b3f6430da45708598bc72e81e2f75b6dc (diff) | |
| parent | 425709eddaa30a39539c4232884272d8064daf4a (diff) | |
Merge pull request #60 from mys007/norms_gcfix
Fix normalization modules leak
| -rw-r--r-- | SpatialCrossMapLRN.lua | 2 | ||||
| -rw-r--r-- | SpatialDivisiveNormalization.lua | 2 |
2 files changed, 2 insertions, 2 deletions
diff --git a/SpatialCrossMapLRN.lua b/SpatialCrossMapLRN.lua index f5ef26f..a08c7af 100644 --- a/SpatialCrossMapLRN.lua +++ b/SpatialCrossMapLRN.lua @@ -22,7 +22,7 @@ function LRN:resetDescriptors() local function destroyDesc(d) errcheck('cudnnDestroyLRNDescriptor', d[0]); end - ffi.gc(self.LRNDesc, destroyPoolDesc) + ffi.gc(self.LRNDesc, destroyDesc) end function LRN:createIODescriptors(input) diff --git a/SpatialDivisiveNormalization.lua b/SpatialDivisiveNormalization.lua index b9a2a16..50d7923 100644 --- a/SpatialDivisiveNormalization.lua +++ b/SpatialDivisiveNormalization.lua @@ -22,7 +22,7 @@ function DivisiveNorm:resetDescriptors() local function destroyDesc(d) errcheck('cudnnDestroyDivisiveNormDescriptor', d[0]); end - ffi.gc(self.DivisiveNormDesc, destroyPoolDesc) + ffi.gc(self.DivisiveNormDesc, destroyDesc) end function DivisiveNorm:createIODescriptors(input) |