dnn: add Vulkan device check for BackendRegistry

This commit is contained in:
Wu Zhiwen 2018-12-24 10:41:58 +08:00
parent e1f5b886fb
commit be6a837e15

View 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)