Systolic primitive

Hello,

I’m working on Eyeriss architecture, as far as I saw in TVM there is no schedule primitive to group PEs for row wise mapping (Row Stationary dataflow see figure in below). Is there a workaround or any new primitive in development plan?