AIMNet2-WB97M-D3-NPU:aimnet2-wb97m-d3 昇腾 NPU 推理适配 PASS - AtomGit AI社区

isayevlab/aimnet2-wb97m-d3 昇腾 NPU 推理适配

模型：isayevlab/aimnet2-wb97m-d3（分子力场）

昇腾 NPU 适配结论

项目	结果
前向输出	CPU/NPU 一致
精度（rel-L2）	9.369097834344747e-07（< 1% 阈值，PASS）
性能（NPU avg）	4.6114858239889145 ms（batch=1）
稳定性	20 轮 batch=1 逐位一致（PASS）
验收	六件套齐全，CPU/NPU 独立子进程对比

关键适配点

aimnet 库 ensemble 加载。
CPU/NPU 独立子进程对比，rel-L2 < 1%。

运行方法

python3 minimal_repro.py
python3 accuracy_eval.py
python3 benchmark.py
python3 stability.py
python3 inference.py

结果文件

results/accuracy_244.json / benchmark_244.json / stability_244.json / inference_results_244.json

三张真实终端截图

assets/agent_workflow.png / assets/npu_device_call.png / assets/model_result.png