We need the UT for the TF operators prod_force_se_*, prod_virial_se_*, prod_force_se_*_grad and prod_virial_se_*_grad, to ensure that they are correctly implemented. These UTs are supposed to be implemented in source/tests and called by the python unittest.
We already have these UTs for the C++ interface, so these UTs are supposed to be implemented from the C++ UTs.
Deepmd-kit Version, Python Version, Tensorflow Version, GCC Version and Cuda Version.
Summary
We need the UT for the TF operators
prod_force_se_*
,prod_virial_se_*
,prod_force_se_*_grad
andprod_virial_se_*_grad
, to ensure that they are correctly implemented. These UTs are supposed to be implemented insource/tests
and called by the pythonunittest
.We already have these UTs for the C++ interface, so these UTs are supposed to be implemented from the C++ UTs.
Deepmd-kit Version, Python Version, Tensorflow Version, GCC Version and Cuda Version.
v2.0.0-beta0
Details