From b3d74f5b882312863f11025fc38e0ca2b0e8f478 Mon Sep 17 00:00:00 2001 From: soumith Date: Sat, 22 Aug 2015 23:12:10 -0400 Subject: flag to enable or not to enable auto-tuner --- test/benchmark.lua | 5 +++-- test/test.lua | 5 +++++ 2 files changed, 8 insertions(+), 2 deletions(-) (limited to 'test') 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) -- cgit v1.2.3