Improved existing 16->32 fallback. Added performance-based fallback.

1 files changed, 3 insertions, 3 deletions

diff --git a/ffi.lua b/ffi.lua
index d5b5f8c..458e382 100644
--- a/ffi.lua
+++ b/ffi.lua
@@ -1614,10 +1614,10 @@ end
 
 -- check cuDNN version
 cudnn.version = tonumber(cudnn.C.cudnnGetVersion())
-if cudnn.version < 5005 then
-  error('These bindings are for version 5005 or above, '
+if cudnn.version < 5005 or cudnn.version >= 6000 then
+   error('These bindings are for CUDNN 5.x (5005 <= cudnn.version > 6000) , '
         .. 'while the loaded CuDNN is version: ' .. cudnn.version
-           .. '  \nAre you using an older version of CuDNN?')
+           .. '  \nAre you using an older or newer version of CuDNN?')
 end
 
 -- check GPU driver version