Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
path: root/test
diff options
context:
space:
mode:
authorBoris Fomitchev <bfomitchev@nvidia.com>2016-08-03 10:41:54 +0300
committerBoris Fomitchev <bfomitchev@nvidia.com>2016-08-03 13:15:56 +0300
commita33739d6346adb3ea262c03a4ff900cef999d8c8 (patch)
tree652031780e237f181cdd1314139b4190653ca5a4 /test
parentca9021c0b60af9e5f3a5313b8927efcff209ede7 (diff)
parentb812efdd1c3465547f0e5b0e24d2053f95ebfe2e (diff)
Merge remote-tracking branch 'upstream/master' into algo
Diffstat (limited to 'test')
-rw-r--r--test/test.lua15
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'