logo
publist
写文章

简介

该用户还未填写简介

擅长的技术栈

可提供的服务

暂无可提供的服务

TVM基本调度原语

TVM调度原语官方DOC#!/usr/bin/env python# coding: utf-8# 通常有几种方法来计算相同的结果,但是不同的方法会导致不同的局部性和性能。所以TVM要求用户提供如何执行称为Schedule的计算。# Schedule是一组转换计算的集合,它转换程序中的计算循环。# In[1]:#兼容python2.xfrom __future__ import print_fun

#python#后端
TVM扫描和递归核(Scan and Recurrent Kernel)

扫描和递归核(Scan and Recurrent Kernel)#!/usr/bin/env python# coding: utf-8# # 扫描和递归核(Scan and Recurrent Kernel)# 递归计算是神经网络的一种典型模式。# In[1]:from __future__ import absolute_import, print_functionimport tvmimp

#python
TVM外部张量函数(External Tensor Functions)

#!/usr/bin/env python# coding: utf-8# # 外部张量函数(External Tensor Functions)## 虽然TVM支持透明的代码生成,但有时将手工编写的代码合并到流水线中也很有帮助。例如,我们可能想用cuDNN来做一些卷积内核,并定义其余的阶段。## TVM原生支持这些黑盒函数调用。确切地说,TVM支持所有与DLPack兼容的张量函数。这意味着我们可

#python
暂无文章信息