1

我有四个功能 A、B、C、DI 希望交错（（0,0）是左上角）：

A B A B ...
C D C D ...
A B A B ...
C D C D ...
...

所以，我有类似的东西：

Func out;              // why can't this be on the same line as the implementation?
out(x, y) = select((x&1)==0, 
               select((y&1)==0, A(x/2,y/2), C(x/2,y/2)),
               select((y&1)==0, B(x/2,y/2), D(x/2,y/2)) );

有没有更好（更有效）的方法来做到这一点？我注意到 A、B、C、D 也包含一些 select()（它们是非常非线性的函数。）

同样，我在 A 内部有这个结构：

Func choose0, choose1, a0, a1, a2, A;

choose0, choose1, a0, a1, a2 = something complicated...;

A(x,y) = select(choose0(x,y), a0(x,y), select(choose1(x,y), a1(x,y), a2(x,y)));

Halide 会在选择之前生成计算 a0、a1 和 a2 的代码吗？我真的宁愿只为（x，y）的任何特定值计算选择的Func。

4

1 回答 1

3

如果你说：

out.unroll(x, 2)
   .unroll(y, 2)
   .bound(x, 0, input.width())
   .bound(y, 0, input.height());

然后选择应该在第一个示例中消失。你告诉 Halide 展开两倍，并且计数从零开始（所以它知道从 A 开始）。

在第二个示例中，Halide 确实会评估选择的两侧 - 否则矢量化效果不佳。诀窍是通过在某处计算 a* Funcs 使这些评估变得便宜（只是加载）：

a0.compute_at(A, x);
a1.compute_at(A, x);
a2.compute_at(A, x);

我选择在 x 处计算它们，因为这样确定它们是否被使用的条件（选择*（x，y））在它们的计算生命周期内是恒定的（x，y 的单个值）。Halide 应该检测到这一点，并在 if 语句中为您干扰他们的计算。

于 2015-08-24T01:07:09.167 回答