Magic-Mirror/llvm-capstone
1
0
Fork 0
mirror of https://github.com/capstone-engine/llvm-capstone.git synced 2025-02-27 13:57:52 +00:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

[mlir] Support use-def cycles in graph regions during regionDCE

Browse Source 
When deleting operations in DCE, the algorithm uses a post-order walk of
the IR to ensure that value uses were erased before value defs. Graph
regions do not have the same structural invariants as SSA CFG, and this
post order walk could delete value defs before uses.  This problem is
guaranteed to occur when there is a cycle in the use-def graph.

This change stops DCE from visiting the operations and blocks in any
meaningful order.  Instead, we rely on explicitly dropping all uses of a
value before deleting it.

Reviewed By: mehdi_amini, rriddle

Differential Revision: https://reviews.llvm.org/D98919
This commit is contained in:
Andrew Young 2021-03-18 20:06:02 -07:00
parent 270a336ff4
commit f178c13fa8
No known key found for this signature in database
GPG Key ID: 88359788021EEFF3
2 changed files with 23 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
mlir/lib/Transforms/Utils/RegionUtils.cpp
View File

@ -312,21 +312,18 @@ static LogicalResult deleteDeadness(MutableArrayRef<Region> regions,
if (region.empty())
continue;
// We do the deletion in an order that deletes all uses before deleting
// defs.
// MLIR's SSA structural invariants guarantee that except for block
// arguments, the use-def graph is acyclic, so this is possible with a
// single walk of ops and then a final pass to clean up block arguments.
//
// To do this, we visit ops in an order that visits domtree children
// before domtree parents. A CFG post-order (with reverse iteration with a
// block) satisfies that without needing an explicit domtree calculation.
// Delete every operation that is not live. Graph regions may have cycles
// in the use-def graph, so we must explicitly dropAllUses() from each
// operation as we erase it. Visiting the operations in post-order
// guarantees that in SSA CFG regions value uses are removed before defs,
// which makes dropAllUses() a no-op.
for (Block *block : llvm::post_order(&region.front())) {
eraseTerminatorSuccessorOperands(block->getTerminator(), liveMap);
for (Operation &childOp :
llvm::make_early_inc_range(llvm::reverse(block->getOperations()))) {
if (!liveMap.wasProvenLive(&childOp)) {
erasedAnything = true;
childOp.dropAllUses();
childOp.erase();
} else {
erasedAnything |=

17
mlir/test/Transforms/canonicalize-dce.mlir
View File

@ -156,3 +156,20 @@ func @f(
"foo.print"(%t4) : (tensor<4xf32>) -> ()
return
}
// -----
// Test case: Test values with use-def cycles are deleted properly.
// CHECK: func @f()
// CHECK-NEXT: test.graph_region
// CHECK-NEXT: "test.terminator"() : () -> ()
func @f() {
test.graph_region {
%0 = "math.exp"(%1) : (f32) -> f32
%1 = "math.exp"(%0) : (f32) -> f32
"test.terminator"() : ()->()
}
return
}