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