2018年12月28日
TVM设备添加以及代码生成
摘要： 因为要添加的设备是一种类似于GPU的加速卡，TVM中提供了对GPU编译器的各种支持，有openCl，OpenGL和CUDA等，这里我们选取比较熟悉的CUDA进行模仿生成。从总体上来看，TVM是一个多层的结构 从上一个文档（TVM调试）中，基本可以发现，TVM在python这一层提供了相关的设备接口， 阅读全文
posted @ 2018-12-28 16:20 转换无极限 阅读(7370) 评论(2) 推荐(0) 编辑