Loop partitioning related issues

Questions
#1

Hi, all

Can TVM support loop partitioning, and back and forth? Say, I got the following loops.

for j in range(J):
  for i in range(49):
     stmt

Then, I may partition it as follows:

for j in range(J):
  for i1 in range(48):
     stmt1
  for i2 in range(1):
     stmt2

Then, I may further split the first part as usual, e.g.,

for j in range(J):
  for i1_o in range(6):
     for i1_i in range(8):
       stmt1
  for i2 in range(1):
     stmt2

Can this be supported now? Better with TE.