Just inline the one stage into the other one?
EDIT: wait your if statements require variables which are not defined (blockIdx.x andThreadIdx.x) --- [Visit Topic](https://discuss.tvm.apache.org/t/ops-become-slow-when-using-te-var/11486/4) 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/d60e4dc363237bb139bbcb74e23315dca9577f5678b7cb522ee479d4f554ffe6).
