2021年11月12日
TVM自定义修改代码示例
摘要： TVM自定义修改代码示例 一．TVM设备添加代码 因为要添加的设备是一种类似于GPU的加速卡，TVM中提供了对GPU编译器的各种支持，有openCl，OpenGL和CUDA等，这里选取比较熟悉的CUDA进行模仿生成。总体上看，TVM是一个多层的结构。 TVM在python这一层提供了相关的设备接口， 阅读全文
posted @ 2021-11-12 06:22 吴建明wujianming 阅读(294) 评论(0) 推荐(0) 编辑