diff options
author | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-01-26 19:11:58 +0300 |
---|---|---|
committer | Sergey Zagoruyko <zagoruyko2@gmail.com> | 2016-01-26 21:49:50 +0300 |
commit | 66fec143a2afba199f752cb416c962bf5e949eb8 (patch) | |
tree | 19b1e02edb16bd2867a9139cdaf6bae3a23e0d5c | |
parent | c743426124be1d5a99377485dcb3263d249b4344 (diff) |
fix volumetric convolution test
-rw-r--r-- | test/test.lua | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/test/test.lua b/test/test.lua index 2a25f98..d6cdffd 100644 --- a/test/test.lua +++ b/test/test.lua @@ -407,6 +407,11 @@ function cudnntest.VolumetricConvolution_forward_single() mytester:assertlt(ferr, precision_io, torch.typename(gconv) .. ' - i/o forward err ') mytester:assertlt(berr, precision_io, torch.typename(gconv) .. ' - i/o backward err ') end + + test(sconv, gconv) + local gconv = cudnn.convert(sconv, cudnn):cuda() + mytester:asserteq(torch.typename(gconv), 'cudnn.VolumetricConvolution', 'conversion type check') + test(sconv, gconv) end function cudnntest.VolumetricConvolution_backward_single() |