dnn: add Vulkan device check for BackendRegistry

6 years ago · be6a837e15
parent e1f5b886fb
commit be6a837e15
1 changed files with 2 additions and 1 deletions
--- 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)