From be6a837e15c88c2a9cabded3c0ef2632d29fdb3d Mon Sep 17 00:00:00 2001 From: Wu Zhiwen Date: Mon, 24 Dec 2018 10:41:58 +0800 Subject: [PATCH] dnn: add Vulkan device check for BackendRegistry --- modules/dnn/src/dnn.cpp | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/modules/dnn/src/dnn.cpp b/modules/dnn/src/dnn.cpp index 9208b249d0..1df85ab52e 100644 --- a/modules/dnn/src/dnn.cpp +++ b/modules/dnn/src/dnn.cpp @@ -137,7 +137,8 @@ private: backends.push_back(std::make_pair(DNN_BACKEND_OPENCV, DNN_TARGET_CPU)); #ifdef HAVE_VULKAN - backends.push_back(std::make_pair(DNN_BACKEND_VKCOM, DNN_TARGET_VULKAN)); // TODO Add device check + if (haveVulkan()) + backends.push_back(std::make_pair(DNN_BACKEND_VKCOM, DNN_TARGET_VULKAN)); #endif } static inline bool checkIETarget(int target) -- GitLab