快速入门Halide
Halide代码流程:
声明函数,变量,表达式
用变量、表达式等实现算法
使用调度策略对算法进行调度调优
调用函数的realize成员函数,对函数进行实现
写出数据
example 1:
Halide::Func gradient;
// Func对象表示了一个pipeline阶段。它是一个纯函数,定义了每个像素点对应的值。
Halide::Var x, y;
// Var对象是Func的定义域,或者说是Func的参数。它们本身没有任何意义。Var用来索引
Halide::Expr e = x
原创
2020-08-21 15:14:57 ·
988 阅读 ·
0 评论