diff options
author | soumith <soumith@fb.com> | 2015-08-20 00:36:25 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2015-08-20 00:36:25 +0300 |
commit | 0b445feb0c7d27e81f40a14b0ca6c16d01f16b63 (patch) | |
tree | deca8abf19aad91dac2a2db797c3b8fde239c87e /SpatialConvolution.lua | |
parent | 40b6e1371939e19cb5ff21b37035c786679b0d6c (diff) |
shared workspace
Diffstat (limited to 'SpatialConvolution.lua')
-rw-r--r-- | SpatialConvolution.lua | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index 79ec341..a99f551 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -188,7 +188,7 @@ function SpatialConvolution:createIODescriptors(input) algType[0], bufSize:data()) maxBufSize = math.max(maxBufSize, bufSize[1]) - self.extraBuffer = self.extraBuffer or input.new(1) + self.extraBuffer = self.extraBuffer or cudnn.getSharedWorkspace() self.extraBufferSizeInBytes = self.extraBuffer:nElement() * 4 -- float if maxBufSize > self.extraBufferSizeInBytes then self.extraBuffer:resize(math.ceil(maxBufSize/4)) |