diff options
| author | soumith <soumith@gmail.com> | 2015-08-23 06:12:10 +0300 |
|---|---|---|
| committer | soumith <soumith@gmail.com> | 2015-08-23 06:12:10 +0300 |
| commit | b3d74f5b882312863f11025fc38e0ca2b0e8f478 (patch) | |
| tree | bbe7dfbef9ea053381af81d6cd0cf292ee9d4e86 /test | |
| parent | 4c2a0a568a232956f0cbe3c89c5e889df0f0ed94 (diff) | |
flag to enable or not to enable auto-tuner
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) |