Browse Source

feat(dnn): fix wgrad rrconv for compute capability

GitOrigin-RevId: ba8792d7a9
release-1.11.1
Megvii Engine Team 2 years ago
parent
commit
6db4620e6d
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      dnn/test/cuda/region_restricted_convolution.cpp

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

@@ -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()}) {


Loading…
Cancel
Save