Could you print out the lowered code? You can use tvm.lower(s, args) where s is the schedule. Also, if you provide a minimal example to run, I can take a look at it.
Could you print out the lowered code? You can use tvm.lower(s, args) where s is the schedule. Also, if you provide a minimal example to run, I can take a look at it.