diff options
author | soumith <soumith@fb.com> | 2015-04-10 05:13:50 +0300 |
---|---|---|
committer | soumith <soumith@fb.com> | 2015-04-10 05:13:50 +0300 |
commit | 35d4f5df368415c27dda955130bcc01d6234ffe6 (patch) | |
tree | 24ea2d025ba13791d85f97a4cbb2adaab627278f /SpatialSoftMax.lua | |
parent | 48b3b6df88198c28086de74a3d74d4745d507f76 (diff) |
using the new streams API (cudnn does not ovelap compute yet, weird)
Diffstat (limited to 'SpatialSoftMax.lua')
-rw-r--r-- | SpatialSoftMax.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/SpatialSoftMax.lua b/SpatialSoftMax.lua index 97c1e38..adbb044 100644 --- a/SpatialSoftMax.lua +++ b/SpatialSoftMax.lua @@ -58,7 +58,7 @@ local zero = torch.FloatTensor({0}); function SpatialSoftMax:updateOutput(input) self:createIODescriptors(input) errcheck('cudnnSoftmaxForward', - cudnn.handle[cutorch.getDevice()-1], + cudnn.getHandle(), self.algorithm, self.mode, one:data(), self.iDesc[0], input:data(), @@ -71,7 +71,7 @@ function SpatialSoftMax:updateGradInput(input, gradOutput) assert(gradOutput:isContiguous()); self:createIODescriptors(input) errcheck('cudnnSoftmaxBackward', - cudnn.handle[cutorch.getDevice()-1], + cudnn.getHandle(), self.algorithm, self.mode, one:data(), self.oDesc[0], self.output:data(), |