diff options
author | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-08-03 10:41:54 +0300 |
---|---|---|
committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-08-03 13:15:56 +0300 |
commit | a33739d6346adb3ea262c03a4ff900cef999d8c8 (patch) | |
tree | 652031780e237f181cdd1314139b4190653ca5a4 /test | |
parent | ca9021c0b60af9e5f3a5313b8927efcff209ede7 (diff) | |
parent | b812efdd1c3465547f0e5b0e24d2053f95ebfe2e (diff) |
Merge remote-tracking branch 'upstream/master' into algo
Diffstat (limited to 'test')
-rw-r--r-- | test/test.lua | 15 |
1 files changed, 6 insertions, 9 deletions
diff --git a/test/test.lua b/test/test.lua index 3d67c9d..8ed2bab 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.fastest = true -- run manual auto-tuner +-- cudnn.benchmark = true -- run manual auto-tuner cudnn.verbose = true --end @@ -1527,19 +1527,16 @@ for i=1,cutorch.getDeviceCount() do print('Running test on device: #' .. i .. ' : ' .. prop.name) 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' |