class nbla::CudaCachedHostArray

class CudaCachedHostArray : public CpuArray

Array allocated on host with a CudaHostMemory obtained by Cuda::pinned_allocator().

Public Functions

explicit CudaCachedHostArray(const Size_t size, dtypes dtype, const Context &ctx, const AllocatorMemoryPtr mem = nullptr, const Size_t offset = 0)

Constructor.

Parameters:

  • size – Length of array.

  • dtype – Data type.

  • ctx – Context.