Welcome to mirror list, hosted at ThFree Co, Russian Federation.

github.com/soumith/cudnn.torch.git - Unnamed repository; edit this file 'description' to name the repository.
summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorBoris Fomitchev <bfomitchev@nvidia.com>2016-09-15 11:35:15 +0300
committerBoris Fomitchev <bfomitchev@nvidia.com>2016-09-15 11:35:15 +0300
commitb146c809cab5a943b802002822f72c3ea6ed0def (patch)
tree2b30fa78f146b3786e69a1f1c3a75fd114fbd7da
parentd09d4e2efe4266eb5165d9d4f61560f3bd46c676 (diff)
debug print fixed
-rw-r--r--find.lua3
1 files changed, 1 insertions, 2 deletions
diff --git a/find.lua b/find.lua
index 5212ceb..b6986c6 100644
--- a/find.lua
+++ b/find.lua
@@ -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]