Unverified Commit efa2baee authored by Juha Reunanen's avatar Juha Reunanen Committed by GitHub
Browse files

Problem: Error while calling cudnnGetConvolutionBackwardFilterWorkspaceSize(...

Problem: Error while calling cudnnGetConvolutionBackwardFilterWorkspaceSize( context(), descriptor(data), descriptor(dest_desc), (const cudnnConvolutionDescriptor_t)conv_handle, (const cudnnFilterDescriptor_t)filter_handle, (cudnnConvolutionBwdFilterAlgo_t)backward_filters_algo, &backward_filters_workspace_size_in_bytes) in file C:\a\1\s\3rdparty\dlib\dlib\cuda\cudnn_dlibapi.cpp:1048. code: 9, reason: CUDNN_STATUS_NOT_SUPPORTED (#2527)

Solution: Do not apply the cuDNN 5.1 workaround if we have cuDNN version greater than or equal to 7
parent 1ccd03fe
...@@ -912,6 +912,7 @@ namespace dlib ...@@ -912,6 +912,7 @@ namespace dlib
&backward_filters_best_algo)); &backward_filters_best_algo));
#endif #endif
#if CUDNN_MAJOR < 7
// cuDNN 5.1 has a bug that causes // cuDNN 5.1 has a bug that causes
// cudnnGetConvolutionBackwardFilterAlgorithm() to pick the winograd // cudnnGetConvolutionBackwardFilterAlgorithm() to pick the winograd
// algorithm even for cases where cuDNN doesn't support it, leading to // algorithm even for cases where cuDNN doesn't support it, leading to
...@@ -926,7 +927,7 @@ namespace dlib ...@@ -926,7 +927,7 @@ namespace dlib
backward_filters_best_algo = CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0; backward_filters_best_algo = CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0;
} }
backward_filters_algo = backward_filters_best_algo; backward_filters_algo = backward_filters_best_algo;
#endif
// Save this algorithm selection in the cache // Save this algorithm selection in the cache
config_to_algo_cache[cache_key] = std::make_tuple(forward_algo, backward_data_algo, backward_filters_algo); config_to_algo_cache[cache_key] = std::make_tuple(forward_algo, backward_data_algo, backward_filters_algo);
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment