================ ---------------- Jianhui-Li wrote: Yes. the inst_data must be contiguous after reshape, so the inner dims must fill up if the distribution unit size is smaller than the original inst_data. https://github.com/llvm/llvm-project/pull/201496