feat(dnn): fix wgrad rrconv for compute capability

GitOrigin-RevId: ba8792d7a9e8c0f424d00606c34c9d800727d239

feat(dnn): fix wgrad rrconv for compute capability
GitOrigin-RevId: ba8792d7a9e8c0f424d00606c34c9d800727d239
6db4620e · Megvii Engine Team · 4e9b1c4e · 6db4620e
隐藏空白更改
内联并排

Showing with 2 addition and 0 deletion

dnn/test/cuda/region_restricted_convolution.cpp dnn/test/cuda/region_restricted_convolution.cpp +2 -0

未找到文件。
--- 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<RegionRestrictedConvolutionBackwardFilter> checker(handle_cuda());

    for (auto dt : std::vector<DType>{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<RegionRestrictedConvolutionBackwardFilter> checker(handle_cuda());

    for (auto dt : std::vector<DType>{dtype::Int32(), dtype::Uint8()}) {