From ca9021c0b60af9e5f3a5313b8927efcff209ede7 Mon Sep 17 00:00:00 2001 From: Boris Fomitchev Date: Tue, 2 Aug 2016 23:50:07 -0700 Subject: Refactoring cudnnFind --- test/test.lua | 15 ++++++++------- 1 file changed, 8 insertions(+), 7 deletions(-) (limited to 'test') diff --git a/test/test.lua b/test/test.lua index aef4daa..3d67c9d 100644 --- a/test/test.lua +++ b/test/test.lua @@ -1516,7 +1516,7 @@ mytester = torch.Tester() mytester:add(cudnntest) -- if torch.random(1,2) == 1 then --- cudnn.benchmark = true -- run manual auto-tuner + cudnn.fastest = true -- run manual auto-tuner cudnn.verbose = true --end @@ -1529,21 +1529,22 @@ for i=1,cutorch.getDeviceCount() do cutorch.setDevice(i) --- double tensor may be broken --- print'Testing torch.CudaDoubleTensor' --- torch.setdefaulttensortype('torch.DoubleTensor') --- testparams = testparams_double --- mytester:run() print'Testing torch.CudaTensor' testparams = testparams_float mytester:run() +-- double tensor may be broken +-- print'Testing torch.CudaDoubleTensor' +-- torch.setdefaulttensortype('torch.DoubleTensor') +-- testparams = testparams_double +-- mytester:run() + +-- half tensor is broken on Pascal print'Testing torch.CudaHalfTensor' testparams = testparams_half mytester:run() - end os.execute('rm -f modelTemp.t7') -- cgit v1.2.3