File gpu_rocm.h
↰ Parent directory (source/lib/include
)
Definition (source/lib/include/gpu_rocm.h
)
Includes
assert.h
errors.h
(File errors.h)hip/hip_runtime.h
stdio.h
vector
Namespaces
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_device_to_host(FPTYPE *, std::vector<FPTYPE>&)
Template Function deepmd::memcpy_host_to_device(FPTYPE *, std::vector<FPTYPE>&)