Merge pull request #13517 from wzw-intel:check_vulkan

6 years ago · b363489702
parent e1f5b886fb be6a837e15
commit b363489702
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)