diff options
author | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-09-15 11:35:15 +0300 |
---|---|---|
committer | Boris Fomitchev <bfomitchev@nvidia.com> | 2016-09-15 11:35:15 +0300 |
commit | b146c809cab5a943b802002822f72c3ea6ed0def (patch) | |
tree | 2b30fa78f146b3786e69a1f1c3a75fd114fbd7da | |
parent | d09d4e2efe4266eb5165d9d4f61560f3bd46c676 (diff) |
debug print fixed
-rw-r--r-- | find.lua | 3 |
1 files changed, 1 insertions, 2 deletions
@@ -301,6 +301,7 @@ function find:setupAlgo(layer, findAPI_idx, algSearchMode, params) local extraBuffer, extraBufferSize = cudnn.getSharedWorkspace() local validResults = 0 + local API local perfResults = perfResultsArray[findAPI_idx] -- try to find algo in the cache first cachedAlgo = self:lookup(layer, findAPI_idx) @@ -323,9 +324,7 @@ function find:setupAlgo(layer, findAPI_idx, algSearchMode, params) local function callCudnn(layer) local ret = 0 - local API validResults = 0 - if cudnn.benchmark or cudnn.fastest then if cudnn.useFindEx then API = findExAlgos[findAPI_idx] |