dnn: add Vulkan device check for BackendRegistry
authorWu Zhiwen <zhiwen.wu@intel.com>
Mon, 24 Dec 2018 02:41:58 +0000 (10:41 +0800)
committerWu Zhiwen <zhiwen.wu@intel.com>
Mon, 24 Dec 2018 02:41:58 +0000 (10:41 +0800)
modules/dnn/src/dnn.cpp

index 9208b24..1df85ab 100644 (file)
@@ -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)