在 Halide Funcs 中设置 Vars 的界限

Question

我有一个这样定义的 3x3 卷积函数

conv(x, y) = 0;
conv(x, y) += kernel(r.x, r.y) * in(x + r.x - 1, y + r.y - 1);

输入缓冲区的大小为 16 x 16
如果我想用填充来执行它，我可以直接做
in = Halide::BoundaryConditions::constant_exterior(in_buffer, 0, 0, 16, 0, 16)
但是我必须在没有填充的情况下执行，所以我试图像这样手动设置函数的界限

conv.bound(x, 1, 14);  
conv.bound(y, 1, 14);

这returns一条错误信息

Error:
Bounds given for convolution in y (from 1 to 14) do not cover required region (from 0 to 15)

我应该如何设置 Func 中 Var 的边界？

Answer 1

我认为您不需要使用 *.bound 函数手动设置边界。试试这个：

Halide::Func conv("conv"), kernelF("kernel"), in("in");
Halide::Var x("x"), y("y");
Halide::RDom r(0, 3, 0, 3,"r");

in = Halide::BoundaryConditions::constant_exterior(in_buffer, 0, 
    0, 16, 0, 16);
kernelF = Halide::BoundaryConditions::constant_exterior(kernel_buffer, 0, 
    0, 3, 0, 3);

conv(x, y) = 0.0f;
conv(x, y) += kernelF(r.x, r.y) * in(x + r.x, y + r.y);

//conv.print_loop_nest();
Halide::Buffer<float_t> outputBuf = conv.realize(14, 14);

看，我们可以直接在 *.realize() 参数中设置边界，即 14=16-3+1;另外，请注意卷积锚位于内核的 top-left 处。

在 Halide Funcs 中设置 Vars 的界限

Setting Bounds on Vars in Halide Funcs

halide