X-Git-Url: http://review.tizen.org/git/?a=blobdiff_plain;f=modules%2Fdnn%2Ftest%2Ftest_common.hpp;h=1202511e541264ec660d840698782d5c2b3f84c2;hb=613c12e59015f4bd7909916ceee195edd7ef88d0;hp=bbbdad4a2fc96f08331e97008944a3b4744bc649;hpb=2df773656265d0349ff0416440f0aff1dabbf79e;p=platform%2Fupstream%2Fopencv.git diff --git a/modules/dnn/test/test_common.hpp b/modules/dnn/test/test_common.hpp index bbbdad4..1202511 100644 --- a/modules/dnn/test/test_common.hpp +++ b/modules/dnn/test/test_common.hpp @@ -18,6 +18,8 @@ #define CV_TEST_TAG_DNN_SKIP_IE_2018R5 "dnn_skip_ie_2018r5" #define CV_TEST_TAG_DNN_SKIP_IE_2019R1 "dnn_skip_ie_2019r1" #define CV_TEST_TAG_DNN_SKIP_IE_2019R1_1 "dnn_skip_ie_2019r1_1" +#define CV_TEST_TAG_DNN_SKIP_IE_2019R2 "dnn_skip_ie_2019r2" +#define CV_TEST_TAG_DNN_SKIP_IE_2019R3 "dnn_skip_ie_2019r3" #define CV_TEST_TAG_DNN_SKIP_IE_OPENCL "dnn_skip_ie_ocl" #define CV_TEST_TAG_DNN_SKIP_IE_OPENCL_FP16 "dnn_skip_ie_ocl_fp16" #define CV_TEST_TAG_DNN_SKIP_IE_MYRIAD_2 "dnn_skip_ie_myriad2" @@ -26,6 +28,9 @@ #define CV_TEST_TAG_DNN_SKIP_VULKAN "dnn_skip_vulkan" +#define CV_TEST_TAG_DNN_SKIP_CUDA "dnn_skip_cuda" +#define CV_TEST_TAG_DNN_SKIP_CUDA_FP16 "dnn_skip_cuda_fp16" +#define CV_TEST_TAG_DNN_SKIP_CUDA_FP32 "dnn_skip_cuda_fp32" namespace cv { namespace dnn { CV__DNN_INLINE_NS_BEGIN @@ -86,7 +91,8 @@ testing::internal::ParamGenerator< tuple > dnnBackendsAndTarget bool withInferenceEngine = true, bool withHalide = false, bool withCpuOCV = true, - bool withVkCom = true + bool withVkCom = true, + bool withCUDA = true ); @@ -106,7 +112,7 @@ public: static void getDefaultThresholds(int backend, int target, double* l1, double* lInf) { - if (target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) + if (target == DNN_TARGET_CUDA_FP16 || target == DNN_TARGET_OPENCL_FP16 || target == DNN_TARGET_MYRIAD) { *l1 = 4e-3; *lInf = 2e-2; @@ -156,6 +162,12 @@ public: expectNoFallbacks(net); } + void expectNoFallbacksFromCUDA(Net& net) + { + if (backend == DNN_BACKEND_CUDA) + expectNoFallbacks(net); + } + protected: void checkBackend(Mat* inp = 0, Mat* ref = 0) {