
Halide
农夫山泉2号
关于交流可以发邮件:778961303@qq.com
展开
-
【嵌入式AI】——autokernel真的能比手动优化更强吗?
出处:openai code:https://github.com/OAID/AutoKernel 1. 简介 随着人工智能的普及,深度学习网络的不断涌现,为了让各硬件(CPU, GPU, NPU,…)能够支持深度学习应用,各硬件芯片需要软件库去支持高性能的深度学习张量运算。目前,这些高性能计算库主要由资深HPC工程师(高性能计算优化工程师)进行开发,为了加快开发进程,缩短深度学习应用落地周期,自动化算子优化是一个趋势。 AutoKernel是由OPEN AI LAB提出的高性能算子自动优化工具,可以.原创 2020-12-03 14:37:06 · 967 阅读 · 1 评论 -
【Halide】——能用if/else做分支选择吗?
参考:stackoverflow 问题: 编写halide程序的时候,想像c++中一样,用if、else做流程控制 Code void generate() { if (stride == 1) { int temp = 1; } else { int temp = 2; } } 说明: 这样是不行的,因为halide虽然是用c++来写的,但是,写的halide(.cc文件)原创 2020-12-04 09:59:48 · 311 阅读 · 0 评论