diff options
| author | Soumith Chintala <soumith@gmail.com> | 2015-10-09 21:59:47 +0300 |
|---|---|---|
| committer | Soumith Chintala <soumith@gmail.com> | 2015-10-09 21:59:47 +0300 |
| commit | ae62fd12fd8bd7d53569855b68d70a2f320ab2f0 (patch) | |
| tree | 51600a57db39d9887d9ad4d253b7db24bb5fe339 /SpatialConvolution.lua | |
| parent | 609f46427f126f21a7a5e1825c411226ff280f65 (diff) | |
| parent | b2c098ea4ba613afa1746140f8f2b8bf22a7a560 (diff) | |
Merge pull request #64 from borisfom/master
Full-fledged version of cudnn_v4 integration.
Diffstat (limited to 'SpatialConvolution.lua')
| -rw-r--r-- | SpatialConvolution.lua | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index dda7b52..bbdffe2 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -380,8 +380,7 @@ function SpatialConvolution:updateOutput(input) end -- add bias - errcheck('cudnnAddTensor_v2', cudnn.getHandle(), - 'CUDNN_ADD_SAME_C', + errcheck('cudnnAddTensor', cudnn.getHandle(), one:data(), self.biasDesc[0], self.bias:data(), one:data(), self.oDescForBias[0], self.output:data()) |