From a0231a79203edee3fc126d386c80cbaf4ae7136c Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 25 Aug 2021 15:59:21 +0800 Subject: [PATCH] fix(dnn/cuda): fix algo matmul for conv bwd filter fix fastrun workspace size not available exception and device OOM error caused by the incorrect workspace size calculation of algo matmul of conv bwd filter GitOrigin-RevId: de96b4fe117ed9691d4a5555261e9f10fa8d2ae4 --- dnn/src/cuda/convolution/backward_filter/matmul.cpp | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dnn/src/cuda/convolution/backward_filter/matmul.cpp b/dnn/src/cuda/convolution/backward_filter/matmul.cpp index 7d0bbeec..4c4a2253 100644 --- a/dnn/src/cuda/convolution/backward_filter/matmul.cpp +++ b/dnn/src/cuda/convolution/backward_filter/matmul.cpp @@ -21,11 +21,11 @@ using namespace cuda; namespace { std::pair sub_opr_config( - const ConvolutionBackwardDataImpl::CanonizedFilterMeta& fm, + const ConvolutionBackwardFilterImpl::CanonizedFilterMeta& fm, const TensorLayout& src_layout, const TensorLayout& diff_layout, const TensorLayout& grad_layout, const ConvolutionBackwardFilterImpl* opr) { - size_t N = grad_layout.shape[0], IC = fm.icpg, + size_t N = src_layout.shape[0], IC = fm.icpg, OC = fm.ocpg, OH = diff_layout.shape[2], OW = diff_layout.shape[3], FH = fm.spatial[0], FW = fm.spatial[1];