diff options
Diffstat (limited to 'Pointwise.lua')
-rw-r--r-- | Pointwise.lua | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/Pointwise.lua b/Pointwise.lua index bf1358a..9b3e71a 100644 --- a/Pointwise.lua +++ b/Pointwise.lua @@ -28,7 +28,7 @@ function Pointwise:updateOutput(input) self:createIODescriptors(input) if self.inplace then self.output = input end errcheck('cudnnActivationForward', - cudnn.handle[cutorch.getDevice()-1], self.mode, + cudnn.getHandle(), self.mode, one:data(), self.iDesc[0], input:data(), zero:data(), @@ -46,7 +46,7 @@ function Pointwise:updateGradInput(input, gradOutput) self:createIODescriptors(input) if self.inplace then self.output = input; self.gradInput = gradOutput end errcheck('cudnnActivationBackward', - cudnn.handle[cutorch.getDevice()-1], self.mode, + cudnn.getHandle(), self.mode, one:data(), self.iDesc[0], self.output:data(), self.iDesc[0], gradOutput:data(), |