Template Function deepmd::prod_virial_grad_r_gpu

• Defined in File prod_virial_grad.h

Function Documentation

template<typename FPTYPE>
void deepmd::prod_virial_grad_r_gpu(FPTYPE *grad_net, const FPTYPE *grad, const FPTYPE *env_deriv, const FPTYPE *rij, const int *nlist, const int nloc, const int nnei)