cudabufferimpl.h

Go to the documentation of this file.

 
#pragma once
 
#include <promeki/namespace.h>
#include <promeki/config.h>
 
#if PROMEKI_ENABLE_CUDA
 
#include <promeki/bufferimpl.h>
#include <promeki/hostbufferimpl.h>
#include <promeki/buffercommand.h>
#include <promeki/error.h>
 
PROMEKI_NAMESPACE_BEGIN
 
class CudaHostBufferImpl : public HostMappedBufferImpl {
        public:
                CudaHostBufferImpl(const MemSpace &ms, size_t bytes, size_t align);
 
                ~CudaHostBufferImpl() override;
 
                CudaHostBufferImpl *_promeki_clone() const override;
};
 
class CudaDeviceBufferImpl : public BufferImpl {
        public:
                CudaDeviceBufferImpl(const MemSpace &ms, size_t bytes, size_t align);
 
                ~CudaDeviceBufferImpl() override;
 
                MemSpace memSpace() const override { return _memSpace; }
                size_t   allocSize() const override { return _allocSize; }
                size_t   align() const override { return _align; }
 
                void *mappedHostData() const override { return nullptr; }
 
                void *devicePtr() const { return _devicePtr; }
 
                BufferRequest mapAcquire(MemDomain domain, MapFlags flags) override;
                BufferRequest mapRelease(MemDomain domain) override;
 
                Error fill(char value, size_t offset, size_t bytes) override;
 
                Error copyFromHost(const void *src, size_t bytes, size_t offset) override;
 
                CudaDeviceBufferImpl *_promeki_clone() const override;
 
        private:
                void    *_devicePtr = nullptr;
                size_t   _allocSize = 0;
                size_t   _align = 0;
                MemSpace _memSpace;
};
 
PROMEKI_NAMESPACE_END
 
#endif // PROMEKI_ENABLE_CUDA