NextCloud/dlib
1
0
Fork 0
mirror of https://github.com/davisking/dlib.git synced 2024-11-01 10:14:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

Fix Cudnn test project bug (#2634)

Browse Source
This commit is contained in:
Mohamed Isse 2022-08-04 18:26:16 -05:00 committed by GitHub
parent 69665eb0f7
commit e2fa5f1e3a
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
dlib/cuda/cudnn_dlibapi.cpp
View File

@ -788,7 +788,7 @@ namespace dlib
select_best_algorithms (
const tensor& data,
const tensor_descriptor& dest_desc,
allow_cache_use allow_cache_use
allow_cache_use allow_cache_use_
)
{
// Calling the cuDNN "find the best algorithm" functions is really slow. So we keep a
@ -800,7 +800,7 @@ namespace dlib
// the cache.
const auto cache_key = std::make_tuple(stride_y, stride_x, padding_y, padding_x, filters_nr, filters_nc);
const auto iter = config_to_algo_cache.find(cache_key);
if (iter != config_to_algo_cache.end() && allow_cache_use == allow_cache_use::yes)
if (iter != config_to_algo_cache.end() && allow_cache_use_ == allow_cache_use::yes)
{
std::tie(forward_algo, backward_data_algo, backward_filters_algo) = iter->second;
return;