Fix convolution backward filter cache when using cudnn 7 and above (#…

…2828) * Fix convolution backward filter cache when using cudnn * Apply suggestions from code review --------- Co-authored-by: Davis E. King <[email protected]>
davisking · Jul 17, 2023 · 3f9c5c1 · 3f9c5c1
1 parent 7736832
commit 3f9c5c1
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dlib/cuda/cudnn_dlibapi.cpp b/dlib/cuda/cudnn_dlibapi.cpp
@@ -927,8 +927,8 @@ namespace dlib
             {
                 backward_filters_best_algo = CUDNN_CONVOLUTION_BWD_FILTER_ALGO_0;
             }
-            backward_filters_algo = backward_filters_best_algo;
 #endif
+	    backward_filters_algo = backward_filters_best_algo;
 
             // Save this algorithm selection in the cache
             config_to_algo_cache[cache_key] = std::make_tuple(forward_algo, backward_data_algo, backward_filters_algo);