dnn: add Vulkan device check for BackendRegistry

be6a837e · Wu Zhiwen · e1f5b886 · be6a837e
Commit be6a837e authored Dec 24, 2018 by Wu Zhiwen
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

dnn.cpp modules/dnn/src/dnn.cpp +2 -1

No files found.
--- 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)