I see. I feel like your intention is to move block "C" under certain loops above block "B". Is that correct. If so, you may use `reverse_compute_at` in your particular case.
try this: ```python def test_compute_at2(): sch = tir.Schedule(compute_at_call_extern, debug_mask="all") print(sch.mod.script()) block_b = sch.get_block("B") block_c = sch.get_block("C") loop_i, = sch.get_loops(block_b) sch.reverse_compute_at(block_c, loop_i, preserve_unit_loops=True) # failed here print(sch.mod.script()) ``` --- [Visit Topic](https://discuss.tvm.apache.org/t/a-failed-example-of-using-compute-at-based-on-tvmscript/11489/3) to respond. You are receiving this because you enabled mailing list mode. To unsubscribe from these emails, [click here](https://discuss.tvm.apache.org/email/unsubscribe/25776657c2deafb35dcc63e2387c7ae21eeb63bd39689108281b6bcaf869fe6b).