Commit0cf42b0

committed

Update op_cuda.hpp

1 parentf503890 commit0cf42b0Copy full SHA for 0cf42b0

File tree

+16

-8

lines changed

+16

-8

lines changed

Lines changed: 15 additions & 1 deletion

Original file line number	Diff line number	Diff line change
`@@ -12,7 +12,6 @@`
`12`	`12`	`namespacecv {namespacednn {`
`13`	`13`	`CV__DNN_INLINE_NS_BEGIN`
`14`	`14`
`15`		`-`
`16`	`15`	`voidNet::Impl::initCUDABackend(const std::vector<LayerPin>& blobsToKeep_)`
`17`	`16`	`{`
`18`	`17`	`CV_Assert(preferableBackend == DNN_BACKEND_CUDA);`
`@@ -107,3 +106,18 @@ void Net::Impl::initCUDABackend(const std::vector<LayerPin>& blobsToKeep_)`
`107`	`106`	`CV__DNN_INLINE_NS_END`
`108`	`107`	`}}// namespace cv::dnn`
`109`	`108`	`#endif// HAVE_CUDA`
	`109`	`+`
	`110`	`+namespacecv {namespacednn {`
	`111`	`+`
	`112`	`+boolhaveCUDA()`
	`113`	`+{`
	`114`	`+#ifdef HAVE_CUDA`
	`115`	`+int dev =0;`
	`116`	`+staticbool ret = (cudaGetDevice(&dev) == cudaSuccess);`
	`117`	`+return ret;`
	`118`	`+#else`
	`119`	`+returnfalse;`
	`120`	`+#endif`
	`121`	`+}`
	`122`	`+`
	`123`	`+}}// namespace cv::dnn`

Lines changed: 1 addition & 7 deletions

Original file line number	Diff line number	Diff line change
`@@ -29,13 +29,7 @@ namespace cv { namespace dnn {`
`29`	`29`	`return id == DNN_TARGET_CUDA_FP16 \|\| id == DNN_TARGET_CUDA;`
`30`	`30`	`}`
`31`	`31`
`32`		`-constexprboolhaveCUDA() {`
`33`		`-#ifdef HAVE_CUDA`
`34`		`-returntrue;`
`35`		`-#else`
`36`		`-returnfalse;`
`37`		`-#endif`
`38`		`- }`
	`32`	`+boolhaveCUDA();`
`39`	`33`
`40`	`34`	`#ifdef HAVE_CUDA`
`41`	`35`	`namespacecuda4dnn {namespacecsl {`

Comments

(0)