third_party/opencv
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Merge pull request #18386 from JulienMaille:patch-1

Browse Source 
* Make sure there is a cuda device before getting it

* Update init.hpp
This commit is contained in:
Julien 2020-09-23 11:15:02 +02:00 committed by GitHub
parent f584c6d723
commit 48ddb53332
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 6 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
modules/dnn/src/cuda4dnn/init.hpp
View File

@ -57,6 +57,9 @@ namespace cv { namespace dnn { namespace cuda4dnn {
bool isDeviceCompatible() bool isDeviceCompatible()
{ {
if (getDeviceCount() <= 0)
return false;
int device_id = getDevice(); int device_id = getDevice();
if (device_id < 0) if (device_id < 0)
return false; return false;
@ -77,6 +80,9 @@ namespace cv { namespace dnn { namespace cuda4dnn {
bool doesDeviceSupportFP16() bool doesDeviceSupportFP16()
{ {
if (getDeviceCount() <= 0)
return false;
int device_id = getDevice(); int device_id = getDevice();
if (device_id < 0) if (device_id < 0)
return false; return false;