DeepMD-kit 结合 LAMMPS 时出现 Segmentation Fault 问题的分析与解决

问题描述

在使用 DeepMD-kit 与 LAMMPS 结合进行分子动力学模拟时，当同时满足以下三个条件时会出现段错误(Segmentation Fault)：

1. 使用 hybrid/scaled pair_style 组合 DeepMD 模型和经验势函数
2. 采用 atom_style full 并包含键/角信息的 data 文件
3. 使用 kspace_style pppm 进行长程静电计算

单独使用其中任意两个条件时模拟可以正常运行，但当三者同时出现时就会导致程序崩溃。

技术背景

DeepMD-kit 是一种基于深度学习的分子动力学模拟工具，可以与 LAMMPS 结合使用。在混合势函数模拟中，开发者经常需要将深度学习势函数与传统力场结合使用，特别是对于包含静电相互作用的体系。

问题根源分析

通过调试和代码分析，发现问题出在邻居列表的处理上。具体表现为：

1. 在计算过程中，邻居列表出现了负值索引（-2147477131）
2. 这个负值索引导致程序在尝试访问非法内存地址时崩溃
3. 问题与 NEIGHMASK 的应用有关 - 在构建邻居列表时未正确处理掩码

解决方案

该问题已在 DeepMD-kit 的最新版本中修复，主要修改包括：

1. 在计算邻居列表时正确应用 NEIGHMASK
2. 增加了对邻居列表索引的合法性检查
3. 优化了数据重排(shuffle)过程中的边界处理

实际应用建议

对于需要使用混合势函数进行模拟的用户，建议：

1. 更新到包含此修复的最新版本 DeepMD-kit
2. 在构建复杂体系时，逐步测试各功能模块的组合
3. 对于包含键/角约束的体系，确保力场参数与深度学习势函数的兼容性
4. 在启用长程静电计算时，注意设置合适的截断半径和精度参数

总结

这个问题的解决使得 DeepMD-kit 能够更稳定地支持复杂体系的混合势函数模拟，特别是对于需要同时考虑键角约束和长程静电作用的体系。开发者现在可以更灵活地组合深度学习势函数与传统力场方法，拓展了 DeepMD-kit 在材料模拟和生物分子模拟中的应用范围。