- 博客(12)
- 收藏
- 关注
RSS订阅翻译 TVM外部张量函数(External Tensor Functions)
#!/usr/bin/env python# coding: utf-8# # 外部张量函数(External Tensor Functions)# # 虽然TVM支持透明的代码生成，但有时将手工编写的代码合并到流水线中也很有帮助。例如，我们可能想用cuDNN来做一些卷积内核，并定义其余的阶段。# # TVM原生支持这些黑盒函数调用。确切地说，TVM支持所有与DLPack兼容的张量函数。这意味着我们可以调用任何具有POD类型(pointer、int、float)或指向DLTensor的指针作为
2021-02-02 22:27:48
466
翻译 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 tvmimport tvm.testingfrom tvm import teimport n
2021-01-27 22:29:30
376
翻译 TVM内置函数和数学函数(Intrinsics and Math Functions)
虽然TVM支持基本的算术操作,但是在很多情况下,我们通常需要更复杂的内置函数.(例如exp指数函数)这些函数依赖于目标系统,在不同的目标平台上可能有不同的名称.下面我们来学习如何调用这些目标特定的函数,以及如何通过tvm的内在API统一接口.from __future__ import absolute_import, print_functionimport tvmfrom tvm import teimport numpy as np直接声明外部数学调用n = te.var("..
2021-01-25 21:35:17
572
原创 python try except finally总结
#!/usr/bin/env python# coding: utf-8# In[1]:a = bprint(a)print("something")"""NameError Traceback (most recent call last)<ipython-input-1-4a5cba587a3a> in <module>----> 1 a = b 2 print(a)
2021-01-20 23:01:44
96
原创 编译器和解释器
编译器编译器是一种计算机语言处理程序，它能够把源程序翻译转换为目标程序，这里的目标程序一般为汇编语言程序或二进制代码程序。需要注意的是，不把整个程序全部翻译成目标程序，目标程序是不能开始运行，且不能产生任何结果的。即编译和运行是两个独立分开的阶段。解释器解释器同样是一种计算机语言处理程序，它接受某种计算机语言的源程序，一个个的获取、分析并执行源程序语句，一旦第一个语句分析结束，便开始运行并且生成结果。也就是说，解释器将“编译”和“运行”结合到了一起。...
2021-01-20 18:44:44
135
翻译 TVM基本调度原语
TVM调度原语官方DOC#!/usr/bin/env python# coding: utf-8# 通常有几种方法来计算相同的结果，但是不同的方法会导致不同的局部性和性能。所以TVM要求用户提供如何执行称为Schedule的计算。# Schedule是一组转换计算的集合，它转换程序中的计算循环。# In[1]:#兼容python2.xfrom __future__ import print_function #在python2.x中也要像python3.x那样使用print（）fr
2021-01-19 21:50:45
275
1
原创 C++类中的静态变量赋初值
C++类中的静态变量虽然默认为私有成员，但是由于它只占用固定的内存，所以可以直接调用赋以初值，代码如下所示。class student{ static int count; string name; string male; int age; int grade;public: student(string name, string male, int age, int grade) { count++; this->name = name; this->male
2021-01-18 22:17:39
5524
翻译 TVM张量表达式
# Get Started with Tensor Expression# # TVM使用一个定义域特定张量表达式来高效地构造核。# In[1]:import tvmimport tvm.testingfrom tvm import teimport numpy as np# In[2]:tgt_host = "llvm"tgt = "cuda"# In[3]:n = te.var("n") #symbolic variable 符号变量A = te.pla.
2021-01-14 21:50:40
325
原创 python“修饰符@“总结
python函数修饰符@的作用是为现有函数增加额外的功能 def without_arg(func): def wrapper(): print("start") func() print("end") return wrapper@without_argdef test(): print("This is a test!") test() """ startThis is a test!end """...
2021-01-13 22:52:42
81
原创 TVM安装--Ubuntu安装TVM
简介TVM是一个用于深度学习系统的编译器堆栈。它旨在缩小以生产力为中心的深度学习框架与以性能和效率为中心的硬件后端之间的差距。TVM与深度学习框架合作，为不同的后端提供端到端编译。换句话说，TVM就是一种将深度学习工作负载部署到硬件的端到段IR（中间表示）堆栈。可以表述为一种把深度学习模型分发到各种硬件设备上的、端到端的解决方案。安装前准备一.building requirementssudo apt-get updatesudo apt-get install -y python3 pytho
2021-01-10 23:24:21
475
原创 Halide(win10&vs2019环境搭建)小试牛刀
什么是HalideHalide,直译为"卤化物"的意思,那么它究竟用来干什么呢?举个栗子:对于简单的for循环二层嵌套,//example1for(int i=0;i<10;i++){ for(int j=0;j<100;j++) { cout<<i+j<<endl; }}和//example2for(int j=0;j<100;j++){ for(int i=0;i<10;i++) { cout<<i+j
2020-11-27 21:21:57
953
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人