class nbla::CudaMemory

class CudaMemory : public Memory

CUDA memory implementation.

A CUDA device memory block allocated by cudaMalloc function is managed by this.

The device passed to constructor is a device id as as string such as “0” and “1”.

Subclassed by nbla::CudaUnifiedMemory