TVM源码阅读记录

最新推荐文章于 2024-08-06 02:02:35 发布

kh815

最新推荐文章于 2024-08-06 02:02:35 发布

阅读量394

点赞数

分类专栏： tvm

本文链接：https://blog.csdn.net/kh815/article/details/115521356

版权

tvm 专栏收录该内容

4 篇文章 0 订阅

订阅专栏

本文介绍了TVM中注册全局操作的方法，特别是如何使用`TVM_REGISTER_GLOBAL`宏结合`.set_body_typed`来为`relay.op.image._make.resize`操作设置类型化的实现。这一过程涉及到TVM的运行时注册机制和类型注解，对于理解和定制TVM的计算图操作至关重要。

摘要由CSDN通过智能技术生成

参考：
https://www.zhihu.com/column/c_1254058636869603328
https://www.zhihu.com/column/c_1093448543469166592

#define TVM_REGISTER_GLOBAL(OpName) \
  TVM_STR_CONCAT(TVM_FUNC_REG_VAR_DEF, __COUNTER__) = ::tvm::runtime::Registry::Register(OpName)
  
TVM_REGISTER_GLOBAL("relay.op.image._make.resize").set_body_typed(MakeResize);
扩展为：

  TVM_STR_CONCAT(TVM_FUNC_REG_VAR_DEF, __COUNTER__) = ::tvm::runtime::Registry::Register(OpName).set_body_typed(MakeResize);