File gpu_rocm.h

↰ Parent directory (source/lib/include)

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

• Program Listing for File gpu_rocm.h

Includes

• assert.h

• errors.h (File errors.h)

• hip/hip_runtime.h

• stdio.h

• string

• vector

Namespaces

• Namespace deepmd

Functions

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

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

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

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

• Function nborAssert(hipError_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