【TVM 教程】外部张量函数
💡
原文中文，约3300字，阅读约需8分钟。
📝
内容提要
Apache TVM 是一个支持 CPU、GPU 和加速芯片的深度学习编译框架。它允许将外部张量函数与 TVM 流水线结合，并支持黑盒函数调用。用户可以通过 te.extern 添加外部函数，以灵活混合自定义代码。
🎯
关键要点
- Apache TVM 是一个支持 CPU、GPU 和加速芯片的深度学习编译框架。
- TVM 支持透明代码生成，可以将手写代码合并到流水线中。
- TVM 支持黑盒函数调用，兼容所有与 DLPack 兼容的张量函数。
- 用户可以通过 te.extern 添加外部数组函数调用，并提供计算函数。
- 可以使用 TVM 的外部 contrib Wrappers 来简化外部调用。
- 可以将 Python 函数注册为外部函数回调，以增强 TVM 的灵活性。
- 总结：TVM 通过 te.extern 调用外部张量函数，使用 contrib wrappers 处理外部调用，并支持前端函数的回调。
➡️
