diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-06-23 22:10:28 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-06-23 22:16:32 +0300 |
commit | cc2d151af59715084a6e837cab873907cbffa22b (patch) | |
tree | 2062438ccc1a26fc5bc3b702b30de77cab842436 /SpatialConvolution.lua | |
parent | a7dc0b62831e04a7fb3c2fb592c4c43c1659e1bb (diff) |
half, double with tests
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 f40a696..a224a6d 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -123,7 +123,7 @@ function SpatialConvolution:createIODescriptors(input) errcheck('cudnnSetConvolutionNdDescriptor', self.convDesc[0], 2, pad:data(), stride:data(), upscale:data(), 'CUDNN_CROSS_CORRELATION', - 'CUDNN_DATA_FLOAT'); + cudnn.configmap[torch.type(self.weight)]); local function destroyConvDesc(d) errcheck('cudnnDestroyConvolutionDescriptor', d[0]); end |