Halide中被两处引用的Func调度报错问题咨询

报错原因 #

你遇到的构建错误核心是Func调度的层级冲突：

• 当给G2设置store_root().compute_at(Q, yi)时，意味着G2的存储被放在根层级，但计算却嵌套在Q的yi循环内部。
• 由于G2同时被Q（compute_root()根层级）和M引用，当M尝试访问G2时，Halide无法保证G2已经在Q的循环中完成计算，或者M的调度层级无法访问到根层级存储的G2数据，从而触发调度合法性检查失败，导致构建报错。
• 而G2.compute_root()让计算和存储都在根层级完成，所有引用它的Func都能正常访问，因此不会报错，但性能不如原调度。

可行解决办法 #

1. 为G2创建多版本调度（推荐） #

利用Halide的clone()方法，为Q和M分别生成G2的独立副本，各自配置适配的调度：

// 克隆G2为两个独立实例
Func G2_Q = G2.clone();
Func G2_M = G2.clone();

// 给Q使用的副本保留原高性能调度
G2_Q.store_root().compute_at(Q, yi);
// 给M使用的副本根据M的调度需求设置，比如compute_root()
G2_M.compute_root();

// 修改Q和M的定义，分别引用对应的G2副本
Q(x, yi) = ... G2_Q(...) ...;
M(x, yi) = ... G2_M(...) ...;

这种方式既保留了原调度对Q的性能优化，又避免了多Func引用同一Func时的调度冲突。

2. 调整G2的存储与计算层级匹配 #

如果G2的计算量不大，可以尝试将调度改为：

G2.compute_root().store_at(Q, yi);

让G2在根层级完成计算，但存储嵌套在Q的yi循环中。不过需要验证M是否能正常访问该存储层级的数据，此方法灵活性不如多版本调度。

3. 统一Q和M的调度层级 #

如果业务允许，将M的调度也设置为compute_root()，此时G2.compute_at(Q, yi)的调度可能可以正常工作。但如果M的计算逻辑不适合根层级调度，此方法不适用。

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