Function DP_DeepPotComputeEdgesGPUFloat32

Function DP_DeepPotComputeEdgesGPUFloat32#

Function Documentation#

void DP_DeepPotComputeEdgesGPUFloat32(DP_DeepPot *dp, double *d_atom_energy, double *d_force, double *d_atom_virial, const double *d_coord, const int *d_atype, const int *d_edge_index, const float *d_edge_vec, int nloc, int nedge, const double *fparam, int64_t fparam_size, const double *aparam, int64_t aparam_size, int nall_nodes, const DP_Nlist *comm_nlist)#

Evaluate a device-resident edge graph with FP32 edge vectors.

The pointer and shape contract matches DP_DeepPotComputeEdgesGPU.

Since

API version 28