[RFC] A general task extraction mechanism for auto_scheduler

[xjwang]

Thanks for your quick reply, it make sense. I’m trying to construct the relay IR to separate OPs into different Functions, but can’t pass checking when relay.build. I create a topic here:

Why relay.Function use a relay.Call can't build successfully through relay.build?

Hi, I’m trying to separate OPs into different relay.Function in order to use their own schedule, Function used a relay.Call to call a sub function: fn (%xx: Tensor[(64, 64), int8], %ww: Tensor[(64), int8], %bb: Tensor[(64), int8], %y: Tensor[(64, 64), int8]) { %0 = fn (%x: Tensor[(64, 64), int8], %w: Tensor[(64), int8], %b: Tensor[(64), int8], Primitive=1) { qnn.aie.layer_norm(%x, %w, %b, epsilon=1.52588e-05f, ln_shift=7, affine_shift=8) }; %1 = %0(%xx, %ww, %bb); nn.dense(%1, %y, units…