Template Function deepmd::prod_virial_a_gpu

Defined in File prod_virial.h

Function Documentation

template<typename FPTYPE> void deepmd::prod_virial_a_gpu(FPTYPE *virial, FPTYPE *atom_virial, const FPTYPE *net_deriv, const FPTYPE *env_deriv, const FPTYPE *rij, const int *nlist, const int nloc, const int nall, const int nnei)