diff options
Diffstat (limited to 'test')
-rw-r--r-- | test/benchmark.lua | 5 | ||||
-rw-r--r-- | test/test.lua | 5 |
2 files changed, 8 insertions, 2 deletions
diff --git a/test/benchmark.lua b/test/benchmark.lua index 08218b9..4372502 100644 --- a/test/benchmark.lua +++ b/test/benchmark.lua @@ -28,6 +28,9 @@ iH = (outH-1)*sH+kH print('CUDNN Version: ', tonumber(cudnn.C.cudnnGetVersion())) +-- just auto-tuned by cudnn with CUDNN_CONVOLUTION_FWD_PREFER_FASTEST mode +bench('Forward AutoTuned ', from, to, kH, kW, sH, sW, iH, iW, batchSize) + bench('Forward implicit gemm ', from, to, kH, kW, sH, sW, iH, iW, batchSize, 'CUDNN_CONVOLUTION_FWD_ALGO_IMPLICIT_GEMM', 'CUDNN_CONVOLUTION_BWD_DATA_ALGO_0', @@ -43,8 +46,6 @@ bench('Forward gemm ', from, to, kH, kW, sH, sW, iH, iW, batchSi 'CUDNN_CONVOLUTION_BWD_DATA_ALGO_0', 'CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0') --- just auto-tuned by cudnn with CUDNN_CONVOLUTION_FWD_PREFER_FASTEST mode -bench('Forward AutoTuned ', from, to, kH, kW, sH, sW, iH, iW, batchSize) bench('Forward FFT ', from, to, kH, kW, sH, sW, iH, iW, batchSize, 'CUDNN_CONVOLUTION_FWD_ALGO_FFT', diff --git a/test/test.lua b/test/test.lua index 5c8b31d..c2938de 100644 --- a/test/test.lua +++ b/test/test.lua @@ -764,6 +764,11 @@ math.randomseed(os.time()) mytester = torch.Tester() mytester:add(cudnntest) +if torch.random(1,2) == 1 then + cudnn.benchmark = true -- run manual auto-tuner +end + + for i=1,cutorch.getDeviceCount() do print('Running test on device: ' .. i) cutorch.setDevice(i) |