intel/llvm
1
0
Fork 0
mirror of https://github.com/intel/llvm.git synced 2026-01-26 03:56:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

[mlir] make transform.loop.outline also return the call handle

Browse Source 
Outlining is particularly interesting when the outlined function is
replaced with something else, e.g., a microkernel. It is good to have a
handle to the call in this case.

Reviewed By: springerm

Differential Revision: https://reviews.llvm.org/D149849
This commit is contained in:
Alex Zinenko
2023-05-04 14:13:06 +00:00
parent 3b6bc87520
commit d906426944
6 changed files with 36 additions and 24 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
mlir/test/python/dialects/transform_loop_ext.py
View File

@@ -33,7 +33,7 @@ def loopOutline():
sequence = transform.SequenceOp(transform.FailurePropagationMode.PROPAGATE,
[], transform.OperationType.get("scf.for"))
with InsertionPoint(sequence.body):
loop.LoopOutlineOp(pdl.OperationType.get(), sequence.bodyTarget, func_name="foo")
loop.LoopOutlineOp(transform.AnyOpType.get(), transform.AnyOpType.get(), sequence.bodyTarget, func_name="foo")
transform.YieldOp()
# CHECK-LABEL: TEST: loopOutline
# CHECK: = transform.loop.outline %