diff --git a/dnn/test/cuda/convolution.cpp b/dnn/test/cuda/convolution.cpp index 4a69c8f4870bcd45b847eacdd6dbb14d60d714d7..0115e7e090a686240927c76fbdf31859d0b136e4 100644 --- a/dnn/test/cuda/convolution.cpp +++ b/dnn/test/cuda/convolution.cpp @@ -17,7 +17,6 @@ #include "test/common/convolution.h" #include "test/common/rng.h" #include "test/cuda/benchmark.h" -#include "test/cuda/utils.h" #include "src/cuda/utils.h" @@ -204,7 +203,7 @@ TEST_F(CUDA, CONVOLUTION_BACKWARD_DATA) .set_epsilon(1e-3) .set_param(arg.param) .exec(TensorLayoutArray{filter, dst, src}); - if (!megdnn::test::check_compute_capability(6, 0)) { + if (!cuda::is_compute_capability_required(6, 0)) { src.dtype = dst.dtype = filter.dtype = dtype::Float16(); checker.set_rng(0, &rng) .set_rng(1, &rng)