PaddlePaddle算子注册原理阅读记录
💡
原文中文,约3400字,阅读约需8分钟。
📝
内容提要
PaddlePaddle通过REGISTER_OPERATOR宏来进行算子注册。该宏的定义和使用方法在文章中有详细介绍。
🎯
关键要点
- PaddlePaddle通过REGISTER_OPERATOR宏进行算子注册。
- 以gru_unit为例,注册代码展示了如何使用REGISTER_OPERATOR宏。
- REGISTER_OPERATOR宏的定义确保在全局命名空间中使用。
- 宏展开后的代码包含全局命名空间检查和算子注册逻辑。
- OperatorRegistrar类负责算子的注册,构造函数中完成注册。
- OpInfoMap用于存储算子信息,确保算子不重复注册。
➡️