File gpu_cuda.h

↰ Parent directory (source/lib/include)

Definition (source/lib/include/gpu_cuda.h)

• Program Listing for File gpu_cuda.h

Includes

• assert.h

• cuda_runtime.h

• errors.h (File errors.h)

• stdio.h

• string

• vector

Included By

• File device.h

Namespaces

• Namespace deepmd

Functions

• Template Function deepmd::delete_device_memory

• Function deepmd::DPGetDeviceCount

• Function deepmd::DPSetDevice

• Template Function deepmd::malloc_device_memory(FPTYPE *&, const std::vector<FPTYPE>&)

• Template Function deepmd::malloc_device_memory(FPTYPE *&, const int)

• Template Function deepmd::malloc_device_memory_sync(FPTYPE *&, const FPTYPE *, const int)

• Template Function deepmd::malloc_device_memory_sync(FPTYPE *&, const std::vector<FPTYPE>&)

• Template Function deepmd::memcpy_device_to_host(const FPTYPE *, FPTYPE *, const int)

• Template Function deepmd::memcpy_device_to_host(const FPTYPE *, std::vector<FPTYPE>&)

• Template Function deepmd::memcpy_host_to_device(FPTYPE *, const std::vector<FPTYPE>&)

• Template Function deepmd::memcpy_host_to_device(FPTYPE *, const FPTYPE *, const int)

• Template Function deepmd::memset_device_memory

• Function DPAssert(cudaError_t, const char *, int, bool)

• Function nborAssert(cudaError_t, const char *, int, bool)

Defines

• Define DPErrcheck

• Define GPU_MAX_NBOR_SIZE

• Define gpuDeviceSynchronize

• Define gpuGetLastError

• Define gpuMemcpy

• Define gpuMemcpyDeviceToDevice

• Define gpuMemcpyDeviceToHost

• Define gpuMemcpyHostToDevice

• Define gpuMemset

• Define nborErrcheck