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:
Diffstat (limited to 'test')
-rw-r--r--test/test.lua15
1 files changed, 8 insertions, 7 deletions
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')