[mlir][scf] Add callback to annotate ops during pipelining
authorThomas Raoux <thomasraoux@google.com>
Tue, 15 Feb 2022 20:23:57 +0000 (12:23 -0800)
committerThomas Raoux <thomasraoux@google.com>
Tue, 15 Feb 2022 20:48:01 +0000 (12:48 -0800)
commit0736bbd7e2f7b43a0246945aaff851cff4827682
tree36d89aee337943c90b1802c00fcc559f76cc6bcf
parent8da319fe770b21d342a534bf02d2b88fffe667cc
[mlir][scf] Add callback to annotate ops during pipelining

This allow user to register a callback that can annotate operations
during software pipelining. This allows user potential annotate op to
know what part of the pipeline they correspond to.

Differential Revision: https://reviews.llvm.org/D119866
mlir/include/mlir/Dialect/SCF/Transforms.h
mlir/lib/Dialect/SCF/Transforms/LoopPipelining.cpp
mlir/test/Dialect/SCF/loop-pipelining.mlir
mlir/test/lib/Dialect/SCF/TestSCFUtils.cpp