From 3a03fa7a5052aeb609accfd73c8dd1b6fdb791c1 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Mon, 7 Sep 2020 16:19:40 +0800 Subject: [PATCH] fix(dnn/cuda): disable pascal sass conv2d GitOrigin-RevId: 385d066595a52cdc11ded5e93889ea978709bba4 --- dnn/src/cuda/conv_bias/algo.cpp | 5 +++-- dnn/test/cuda/conv_bias_int8.cpp | 1 + 2 files changed, 4 insertions(+), 2 deletions(-) diff --git a/dnn/src/cuda/conv_bias/algo.cpp b/dnn/src/cuda/conv_bias/algo.cpp index df80af20..57df4902 100644 --- a/dnn/src/cuda/conv_bias/algo.cpp +++ b/dnn/src/cuda/conv_bias/algo.cpp @@ -6,7 +6,8 @@ * * Unless required by applicable law or agreed to in writing, * software distributed under the License is distributed on an - * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. + * "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or + * implied. */ #include "src/cuda/conv_bias/algo.h" @@ -47,7 +48,7 @@ ConvBiasForwardImpl::AlgoPack::AlgoPack() { conv_algos.reserve(conv_algos.size() * 2); //! add gconv algos by AlgoGroupConvGeneral size_t algo_size = conv_algos.size(); - for (size_t i = 3; i < algo_size; ++ i) { + for (size_t i = 3; i < algo_size; ++i) { gconv_refhold.emplace_back(new AlgoGroupConvGeneral(conv_algos[i])); algo2gconv[conv_algos[i]] = gconv_refhold.back().get(); conv_algos.push_back(gconv_refhold.back().get()); diff --git a/dnn/test/cuda/conv_bias_int8.cpp b/dnn/test/cuda/conv_bias_int8.cpp index 256cf15f..58b769be 100644 --- a/dnn/test/cuda/conv_bias_int8.cpp +++ b/dnn/test/cuda/conv_bias_int8.cpp @@ -1076,6 +1076,7 @@ TEST_F(CUDA, CONV_BIAS_INT8_CHWN4_UNROLL_WIDTH_TENSORCORE_1x1_ALGO_2) { } + #if CUDA_VERSION >= 10020 /// \note: we only check several cases and block sizes in megdnn_test, the full /// testcases are written in cutlass repository