[BUG] compressed model produces NaN forces

[njzjz]

Summary

Here I have a system. Predicted forces by an uncompressed model:

Forces predicted by the compressed model:

We can see most of atoms have the same forces, but some of forces are NaN. (Python interface)

Deepmd-kit version, installation way, input file, running commands, error log, etc.

devel, f5b0abb

Steps to Reproduce
I upload the model and data here: 1.zip.
The compress model is too big to upload, but it should be able to compress by dp compress.

dpdata's predict method can be used here.

Further Information, Files, and Links

[njzjz]

UPDATE: I have no issue with CPUs. It should be a problem of GPU ops.

[denghuilu]

UPDATE: I have no issue with CPUs. It should be a problem of GPU ops.

#1575