From 6db4620e6dc95f50d884334fbb96a54e28c718dd Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Sun, 2 Oct 2022 20:30:58 +0800 Subject: [PATCH] feat(dnn): fix wgrad rrconv for compute capability GitOrigin-RevId: ba8792d7a9e8c0f424d00606c34c9d800727d239 --- dnn/test/cuda/region_restricted_convolution.cpp | 2 ++ 1 file changed, 2 insertions(+) diff --git a/dnn/test/cuda/region_restricted_convolution.cpp b/dnn/test/cuda/region_restricted_convolution.cpp index 1e87f474..06857a2d 100644 --- a/dnn/test/cuda/region_restricted_convolution.cpp +++ b/dnn/test/cuda/region_restricted_convolution.cpp @@ -786,6 +786,7 @@ TEST_F(CUDA, REGION_RESTRICTED_CONV_BWD_DATA_FP32_RIN_EQ_ROUT) { } TEST_F(CUDA, REGION_RESTRICTED_CONV_BWD_FILTER_FP32) { + require_compute_capability(6, 1); Checker checker(handle_cuda()); for (auto dt : std::vector{dtype::Int32(), dtype::Uint8()}) { @@ -846,6 +847,7 @@ TEST_F(CUDA, REGION_RESTRICTED_CONV_BWD_FILTER_FP32) { } TEST_F(CUDA, REGION_RESTRICTED_CONV_BWD_FILTER_FP32_RIN_EQ_ROUT) { + require_compute_capability(6, 1); Checker checker(handle_cuda()); for (auto dt : std::vector{dtype::Int32(), dtype::Uint8()}) {