Skip to content

Commit b084b5b

Browse files
authored
Fixup wrong key lookup. (#2048)
This is a follow up to #1948.
1 parent 4b87ec0 commit b084b5b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

lib/cudnn/src/convolution.jl

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -279,7 +279,7 @@ function cudnnConvolutionBwdFilterAlgoPerf(xDesc, x, dyDesc, dy, convDesc, dwDes
279279

280280
key = (xDesc_native, dyDesc_native, convDesc_native)
281281
val = lock(cudnnConvolutionBwdFilterAlgoPerfCacheLock) do
282-
get(cudnnConvolutionBwdFilterAlgoPerfCache, (xDesc, dyDesc, convDesc), nothing)
282+
get(cudnnConvolutionBwdFilterAlgoPerfCache, key, nothing)
283283
end
284284
if val === nothing
285285
requestedAlgoCount = Int(CUDNN_CONVOLUTION_BWD_FILTER_ALGO_COUNT)

0 commit comments

Comments
 (0)