ge - ge技术,学习,经验文章

hh.h.

7 小时前

昇腾CANN ge 仓：图引擎的架构与实战训练大模型的时候，计算图要跑到昇腾NPU上，中间要经过一层转换——从 PyTorch/TensorFlow 的计算图转换成昇腾的图格式。ge（Graph Engine）就是干这个的，它是 CANN 的图编译和运行时引擎，位于第三层（昇腾计算编译层）。这篇文章拆开看 ge 的架构，以及怎么用它做图优化。