Rewrote bias fprop to avoid serialization cuda->host.

1 files changed, 4 insertions, 3 deletions

diff --git a/SpatialConvolutionCUDA.lua b/SpatialConvolutionCUDA.lua
index 6ecaf3f..baa79af 100644
--- a/SpatialConvolutionCUDA.lua
+++ b/SpatialConvolutionCUDA.lua
@@ -37,9 +37,10 @@ end
 
 function SpatialConvolutionCUDA:updateOutput(input)
    input.nn.SpatialConvolutionCUDA_updateOutput(self, input)
-   for i = 1,self.nOutputPlane do
-      self.output[i]:add(self.bias[i])
-   end
+   local biasrep = self.bias:new():resize(self.bias:size(1),1,1,1):expandAs(self.output)
+   self.biasrepc = self.biasrepc or biasrep.new()
+   self.biasrepc:resizeAs(self.output):copy(biasrep)
+   self.output:add(self.biasrepc)
    return self.output
 end