From 10804e55305343615bfd1c36e323fc8096786c46 Mon Sep 17 00:00:00 2001 From: Romeo Valentin Date: Sun, 20 Aug 2023 15:48:43 -0700 Subject: [PATCH] Fixup wrong key lookup. This is a follow up to #1948. --- lib/cudnn/src/convolution.jl | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/lib/cudnn/src/convolution.jl b/lib/cudnn/src/convolution.jl index 0294faf53e..34c587a8ed 100644 --- a/lib/cudnn/src/convolution.jl +++ b/lib/cudnn/src/convolution.jl @@ -279,7 +279,7 @@ function cudnnConvolutionBwdFilterAlgoPerf(xDesc, x, dyDesc, dy, convDesc, dwDes key = (xDesc_native, dyDesc_native, convDesc_native) val = lock(cudnnConvolutionBwdFilterAlgoPerfCacheLock) do - get(cudnnConvolutionBwdFilterAlgoPerfCache, (xDesc, dyDesc, convDesc), nothing) + get(cudnnConvolutionBwdFilterAlgoPerfCache, key, nothing) end if val === nothing requestedAlgoCount = Int(CUDNN_CONVOLUTION_BWD_FILTER_ALGO_COUNT)