Template Function deepmd::prod_env_mat_r_gpu

• Defined in File prod_env_mat.h

Function Documentation

template<typename FPTYPE>
void deepmd::prod_env_mat_r_gpu(FPTYPE *em, FPTYPE *em_deriv, FPTYPE *rij, int *nlist, const FPTYPE *coord, const int *type, const InputNlist &gpu_inlist, int *array_int, unsigned long long *array_longlong, const int max_nbor_size, const FPTYPE *avg, const FPTYPE *std, const int nloc, const int nall, const float rcut, const float rcut_smth, const std::vector<int> sec)