[ModuloSchedule] Pass loop block explicitly to kernel rewriter.
authorHendrik Greving <hgreving@google.com>
Wed, 23 Jun 2021 19:32:04 +0000 (12:32 -0700)
committerHendrik Greving <hgreving@google.com>
Fri, 25 Jun 2021 16:51:22 +0000 (09:51 -0700)
commite15e1417b9fdd628476736c7b1637779752ae3c7
treefe0a123aa1845fca9bb77583a9dc47f3b6261b56
parentb7f24923a3027e66b78b56174e9950c146203104
[ModuloSchedule] Pass loop block explicitly to kernel rewriter.

This change is NFC upstream. We pass in the loop's block to the kernel
rewriter explicitly, instead of assuming it's the loop's top block. This
change is made for downstream targets where this assumption doesn't hold.

Differential Revision: https://reviews.llvm.org/D104811
llvm/lib/CodeGen/ModuloSchedule.cpp