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

Defined in File gpu_rocm.h

Function Documentation

template<typename FPTYPE> void deepmd::malloc_device_memory(FPTYPE *&device, std::vector<FPTYPE> &host)

Read the Docs v: master

Versions: master; latest; stable; v2.2.6; v2.2.5; v2.2.4; v2.2.3; v2.2.2; v2.2.1; v2.2.0; v2.2.0.b0; v2.1.5; v2.1.4; v2.1.3; v2.1.2; v2.1.1; v2.1.0; v2.0.3; v2.0.2; v2.0.1; v2.0.0; v2.0.0.b4; v2.0.0.b3; v2.0.0.b2; v2.0.0.b1; v2.0.0.b0; v2.0.0.a1; v2.0.0.a0; v1.3.3; v1.3.2; v1.3.1; v1.3.0; r1.3; r1.2; devel-20220601; devel-20220531

Downloads: pdf; html; epub

On Read the Docs: Project Home; Builds