tvm的注册机制分析

最新推荐文章于 2022-07-03 10:21:25 发布

种西瓜的大白兔

最新推荐文章于 2022-07-03 10:21:25 发布

阅读量171

点赞数 1

分类专栏： TVM 文章标签： c++

本文链接：https://blog.csdn.net/zhutouying/article/details/123711195

版权

本文分析了TVM的注册机制，包括TVM_REGISTER_TARGET_KIND宏定义的使用，连续调用函数的原因，全局域中的注册以及注册内容的存储位置。重点讨论了静态变量在注册过程中的作用以及TargetKindRegEntry类的功能。

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

TVM_REGISTER_TARGET_KIND 是如何注册的呢？

#define TVM_REGISTER_TARGET_KIND(TargetKindName, DeviceType)      \
  TVM_STR_CONCAT(TVM_TARGET_KIND_REGISTER_VAR_DEF, __COUNTER__) = \
      ::tvm::TargetKindRegEntry::RegisterOrGet(TargetKindName)    \
          .set_name()                                             \
          .set_device_type(DeviceType)                            \
          .add_attr_option<Array<String>>("keys")                 \
          .add_attr_option<String>("tag")                         \
          .add_attr_option<String>("device")                      \
          .add_attr_option<String>("model")                       \
          .add_attr_option<Array<String>>("libs")                 \
          .add_attr_option<Target>("host")                        \
          .add_attr_option<Integer>("from_device")