16 #ifndef TMVA_DNN_ARCHITECTURES_CUDA_CUDABUFFERS 17 #define TMVA_DNN_ARCHITECTURES_CUDA_CUDABUFFERS 20 #include "cuda_runtime.h" 27 template<
typename AFloat>
41 template<
typename AFloat>
77 operator AFloat * ()
const;
96 template<
typename AFloat>
132 operator AFloat * ()
const;
146 template<
typename AFloat>
152 template<
typename AFloat>
TDestructor & operator=(const TDestructor &)=default
std::shared_ptr< AFloat * > fHostPointer
Pointer to the buffer data.
void SetComputeStream(cudaStream_t stream)
cudaStream_t GetComputeStream() const
size_t fOffset
Offset for sub-buffers.
AFloat & operator[](size_t index)
struct TMVA::DNN::TCudaHostBuffer::TDestructor fDestructor
cudaStream_t fComputeStream
cudaStream for data transfer
void operator()(AFloat **devicePointer)
std::shared_ptr< AFloat * > fDevicePointer
Pointer to the buffer data.
TCudaHostBuffer()=default
Abstract ClassifierFactory template that handles arbitrary types.
TCudaHostBuffer GetSubBuffer(size_t offset, size_t size)
Return sub-buffer of the current buffer.
size_t fOffset
Offset for sub-buffers.
cudaStream_t fComputeStream
cudaStream for data transfer