Halide中使用compute_with调度指令报错问题求助

问题原因 #

你遇到的错误是因为Halide默认将未显式调度且使用场景简单的Func（这里是A）内联了。内联后的Func没有独立的计算循环结构，无法通过compute_with()与其他Func（B）的循环绑定合并。

解决方案 #

核心是先解除A的内联状态，再通过compute_with()将其与B的循环绑定，具体步骤如下：

1. 解除A的内联状态：显式为A指定非内联的调度规则（比如compute_at()、compute_root()），让A拥有独立的计算循环。
2. 绑定循环：使用compute_with()将A绑定到B的目标循环维度，确保两者计算域尺寸匹配。

修复后的代码示例 #

假设你的最小复现代码如下，修复后的调度逻辑如下：

#include "Halide.h"
using namespace Halide;

int main() {
    Func A("A"), B("B"), C("C");
    Var x("x");

    // 模拟实际的长计算流程
    A(x) = x + 1; // 替换为你的复杂计算逻辑
    B(x) = x * 2; // 替换为你的复杂计算逻辑
    C(x) = A(x) + B(x);

    // 正确调度：先设置B为compute_root，再处理A的绑定
    B.compute_root();
    // 先让A在B的x维度循环中计算（解除内联），再绑定到B的循环
    A.compute_at(B, x).compute_with(B, x);

    C.realize(10);
    return 0;
}

替代写法 #

如果希望A和B在最顶层root循环合并，也可以用更简洁的写法：

B.compute_root();
A.compute_root().compute_with(B, x);

注意事项 #

• 确保A和B的计算域尺寸完全一致，否则compute_with()会因维度不匹配报错。
• 若涉及多维度循环（比如x、y），需指定具体的循环变量（如compute_with(B, y)），确保绑定到正确的循环层级。

内容的提问来源于stack exchange，提问作者Cordovan