Commit 1bdd86ed authored by Vadim Pisarevsky's avatar Vadim Pisarevsky

Merge pull request #3523 from jet47:fix-cuda-buffer-pool

parents cddee22c 68e08bbe
...@@ -490,6 +490,7 @@ private: ...@@ -490,6 +490,7 @@ private:
friend struct StreamAccessor; friend struct StreamAccessor;
friend class BufferPool; friend class BufferPool;
friend class DefaultDeviceInitializer;
}; };
class CV_EXPORTS Event class CV_EXPORTS Event
......
...@@ -92,26 +92,6 @@ static inline void throw_no_cuda() { CV_Error(cv::Error::StsNotImplemented, "The ...@@ -92,26 +92,6 @@ static inline void throw_no_cuda() { CV_Error(cv::Error::StsNotImplemented, "The
namespace cv { namespace cuda namespace cv { namespace cuda
{ {
class MemoryStack;
class CV_EXPORTS StackAllocator : public GpuMat::Allocator
{
public:
explicit StackAllocator(cudaStream_t stream);
~StackAllocator();
bool allocate(GpuMat* mat, int rows, int cols, size_t elemSize);
void free(GpuMat* mat);
private:
StackAllocator(const StackAllocator&);
StackAllocator& operator =(const StackAllocator&);
cudaStream_t stream_;
MemoryStack* memStack_;
size_t alignment_;
};
class CV_EXPORTS BufferPool class CV_EXPORTS BufferPool
{ {
public: public:
...@@ -120,6 +100,8 @@ namespace cv { namespace cuda ...@@ -120,6 +100,8 @@ namespace cv { namespace cuda
GpuMat getBuffer(int rows, int cols, int type); GpuMat getBuffer(int rows, int cols, int type);
GpuMat getBuffer(Size size, int type) { return getBuffer(size.height, size.width, type); } GpuMat getBuffer(Size size, int type) { return getBuffer(size.height, size.width, type); }
GpuMat::Allocator* getAllocator() const { return allocator_; }
private: private:
GpuMat::Allocator* allocator_; GpuMat::Allocator* allocator_;
}; };
......
This diff is collapsed.
This diff is collapsed.
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment