Amy Zhuang 7ab14b8886 [mlir] Unroll-and-jam loops with iter_args. ...

Unroll-and-jam currently doesn't work when the loop being unroll-and-jammed
or any of its inner loops has iter_args. This patch modifies the
unroll-and-jam utility to support loops with iter_args.

Reviewed By: bondhugula

Differential Revision: https://reviews.llvm.org/D110085

2021-09-28 14:13:27 -07:00

48 KiB

Raw Blame History

View Raw