From 66fec143a2afba199f752cb416c962bf5e949eb8 Mon Sep 17 00:00:00 2001 From: Sergey Zagoruyko Date: Tue, 26 Jan 2016 17:11:58 +0100 Subject: fix volumetric convolution test --- test/test.lua | 5 +++++ 1 file changed, 5 insertions(+) (limited to 'test') 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() -- cgit v1.2.3