Hongtao Yu fa57c7a6a5
[mlir] Extend SCF loopUnrollByFactor to return the result loops (#114573)
There is a need of accessing the resulted epilog loop from the SC loop
unroller. It'd clean and convenient to get that directly from the loop
unroller instead of rescanning the whole function, as discussed in
https://github.com/triton-lang/triton/pull/5027 . I'm changing the
result type of `loopUnrollByFactor` for that.
2024-11-04 15:05:28 -08:00
..