将值输出到不同大小的缓冲区

Question

我需要将 2 个值输出到不同大小的缓冲区。一个与输入大小相同，另一个是宽度的一半和高度的一半。我想过使用元组，但我越界了，因为它在输入大小上进行迭代。这是一个简单的例子：

uint8_t _in[4] = { 1, 2, 3, 4 };
uint8_t _out1[4];
uint8_t _out2[1];

Buffer<uint8_t> in(_in, 2, 2);
Buffer<uint8_t> out1(_out1, 2, 2);
Buffer<uint8_t> out2(_out2, 1, 1);

Var x, y;
Func f;

f(x, y) = {
    in(x,y),
    in(x / 2, y / 2)
};

f.realize({ out1, out2 });

元组不是解决这个问题的正确方法吗？

Answer 1

使用两个输出 Funcs 构造一个 Halide::Pipeline 并在 Pipeline 对象上调用 realize。

大概是这样：

uint8_t _in[4] = { 1, 2, 3, 4 };
uint8_t _out1[4];
uint8_t _out2[1];

Buffer<uint8_t> in(_in, 2, 2);
Buffer<uint8_t> out1(_out1, 2, 2);
Buffer<uint8_t> out2(_out2, 1, 1);

Var x, y;
Func f1, f2;

f1(x, y) = in(x,y);
f2(x, y) = in(x / 2, y / 2);

Pipeline pipeline({f1, f2});

pipeline.realize({ out1, out2 });

将值输出到不同大小的缓冲区

Output values to buffers of different sizes

c++

halide