From 1c4a64b2af0a7c9c9f44ba6dec215b08caacd2a2 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Thu, 30 Apr 2020 19:36:05 +0800 Subject: [PATCH] test(megdnn): skip fp16 test if compute capability less than 60 GitOrigin-RevId: 3377007b875ad1538ea5ecbac2406e6941683ab8 fix(dnn): fix include file redefined error GitOrigin-RevId: 9d52f7ba503ce3a8df6058d450c023bef1e6d59f --- dnn/test/cuda/convolution.cpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/dnn/test/cuda/convolution.cpp b/dnn/test/cuda/convolution.cpp index 4a69c8f48..0115e7e09 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) -- GitLab