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