Merge pull request #25978 from chacha21:cuda_stdallocator

asmorkalov · web-flow · commit 9de2ebbec1c9 · 2024-08-05T10:58:33.000+03:00
Adding getStdAllocator() to cv::cuda::GpuMat
diff --git a/modules/core/include/opencv2/core/cuda.hpp b/modules/core/include/opencv2/core/cuda.hpp
@@ -118,6 +118,7 @@ class CV_EXPORTS_W GpuMat
     //! default allocator
     CV_WRAP static GpuMat::Allocator* defaultAllocator();
     CV_WRAP static void setDefaultAllocator(GpuMat::Allocator* allocator);
+    CV_WRAP static GpuMat::Allocator* getStdAllocator();
 
     //! default constructor
     CV_WRAP explicit GpuMat(GpuMat::Allocator* allocator = GpuMat::defaultAllocator());
diff --git a/modules/core/src/cuda/gpu_mat.cu b/modules/core/src/cuda/gpu_mat.cu
@@ -135,6 +135,7 @@ namespace
 
     DefaultAllocator cudaDefaultAllocator;
     GpuMat::Allocator* g_defaultAllocator = &cudaDefaultAllocator;
+    GpuMat::Allocator* g_stdAllocator = &cudaDefaultAllocator;
 }
 
 GpuMat::Allocator* cv::cuda::GpuMat::defaultAllocator()
@@ -148,6 +149,12 @@ void cv::cuda::GpuMat::setDefaultAllocator(Allocator* allocator)
     g_defaultAllocator = allocator;
 }
 
+GpuMat::Allocator* cv::cuda::GpuMat::getStdAllocator()
+{
+    return g_stdAllocator;
+}
+
+
 /////////////////////////////////////////////////////
 /// create
 
diff --git a/modules/core/src/cuda_gpu_mat.cpp b/modules/core/src/cuda_gpu_mat.cpp
@@ -420,6 +420,11 @@ void cv::cuda::GpuMat::setDefaultAllocator(Allocator* allocator)
     throw_no_cuda();
 }
 
+GpuMat::Allocator* cv::cuda::GpuMat::getStdAllocator()
+{
+    return 0;
+}
+
 void cv::cuda::GpuMat::create(int _rows, int _cols, int _type)
 {
     CV_UNUSED(_rows);

Original file line number	Diff line number	Diff line change
`@@ -420,6 +420,11 @@ void cv::cuda::GpuMat::setDefaultAllocator(Allocator* allocator)`
`420`	`420`	`throw_no_cuda();`
`421`	`421`	`}`
`422`	`422`
	`423`	`+GpuMat::Allocator* cv::cuda::GpuMat::getStdAllocator()`
	`424`	`+{`
	`425`	`+ return 0;`
	`426`	`+}`
	`427`	`+`
`423`	`428`	`void cv::cuda::GpuMat::create(int _rows, int _cols, int _type)`
`424`	`429`	`{`
`425`	`430`	`CV_UNUSED(_rows);`