Mxnet Source Read 2

Mxnet/src/storage/gpu_device_storage.h

Similar to cpu_device_storage.h is defined by class Gpudevicestorage with level two namespaces mxnet and storage, with two static member functions inside

Alloc and free implementations are implemented by invoking the Cuda API Cudamalloc and Cudafree, while the CPU portion is not logged when the memory request and the release error are written with log information.

Personal feelings include the header file is a bit large, can be more detailed only include logging.h only.

At the same time for CUDA return error detection more than a cudaerrorcudartunloading not accepted, the official document gives two value for! Cudasuccess should be treated as errors, and there might be some logic in the side.

Even a hardware-driven error should be wrong, but there is no error in dealing with it.

Mxnet Source Read 2