Fixup wrong key lookup. (#2048)

This is a follow up to #1948.
JuliaGPU · Aug 21, 2023 · b084b5b · b084b5b
1 parent 4b87ec0
commit b084b5b
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)