diff options
| author | Boris Fomitchev <bfomitchev@nvidia.com> | 2015-10-14 03:23:34 +0300 |
|---|---|---|
| committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2015-10-14 03:23:34 +0300 |
| commit | 350243991bf7c1f76648c668df2df6be644a2b31 (patch) | |
| tree | 13415fced8bb26b2dd8ff526b83896f34d27192c /SpatialConvolution.lua | |
| parent | b2c098ea4ba613afa1746140f8f2b8bf22a7a560 (diff) | |
Bugfix by Andrei
Diffstat (limited to 'SpatialConvolution.lua')
| -rw-r--r-- | SpatialConvolution.lua | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/SpatialConvolution.lua b/SpatialConvolution.lua index 5f7e0e0..f84792b 100644 --- a/SpatialConvolution.lua +++ b/SpatialConvolution.lua @@ -151,8 +151,8 @@ function SpatialConvolution:createIODescriptors(input) 1, intt:data(), perfResults) algType[0] = perfResults[0].algo if cudnn.verbose then - print('AutoTuning:', perfResults[0].time, - tonumber(perfResults[0].memory), tonumber(perfResults[0].algo)) + print('cudnnConvolutionFwdAlgoPerf\t\tAutoTuning:', perfResults[0].time, '\t', + tonumber(perfResults[0].memory), '\t', tonumber(perfResults[0].algo)) end else errcheck('cudnnGetConvolutionForwardAlgorithm', @@ -190,8 +190,8 @@ function SpatialConvolution:createIODescriptors(input) 1, intt:data(), perfResults) algType[0] = perfResults[0].algo if cudnn.verbose then - print('AutoTuning:', perfResults[0].time, - tonumber(perfResults[0].memory), tonumber(perfResults[0].algo)) + print('cudnnFindConvolutionBwdFilterAlgoPerf_t\tAutoTuning:', perfResults[0].time, '\t', + tonumber(perfResults[0].memory), '\t', tonumber(perfResults[0].algo)) end else errcheck('cudnnGetConvolutionBackwardFilterAlgorithm', @@ -228,8 +228,8 @@ function SpatialConvolution:createIODescriptors(input) 1, intt:data(), perfResults) algType[0] = perfResults[0].algo if cudnn.verbose then - print('AutoTuning:', perfResults[0].time, - tonumber(perfResults[0].memory), tonumber(perfResults[0].algo)) + print('cudnnConvolutionBwdDataAlgoPerf_t\tAutoTuning:', perfResults[0].time, '\t', + tonumber(perfResults[0].memory), '\t', tonumber(perfResults[0].algo)) end else errcheck('cudnnGetConvolutionBackwardDataAlgorithm', |