如何强制 LLVM 生成单个“ret”？

Question

考虑以下 LLVM IR：

@yyy = external dso_local global i32
@zzz = external dso_local global i64

define void @exec_xxx() {
entry:
  %0 = load i32, i32* @yyy, align 4
  %1 = icmp eq i32 %0, 0
  br i1 %1, label %bb_true, label %bb_false
bb_true:
  store i64 0, i64* @zzz, align 8
  br label %bb_false
bb_false:
  ret void
}

在这里我们看到它有一个ret 。 但是，生成的代码有多个ret ：

exec_xxx:                               # @exec_xxx
        cmp     dword ptr [rip + yyy], 0
        je      .LBB0_1
        ret
.LBB0_1:                                # %bb_true
        mov     qword ptr [rip + zzz], 0
        ret

由于某种原因，在生成的代码中需要一个ret 。

问题：如何强制 LLVM 生成单个“ret”？

Answer 1

LLVMish 对此的回答是写一个通行证。 Passes 是 LLVM 修改代码的方式。

在这种情况下，这将非常简单，您需要一个继承PassInfoMixin<>并重新实现run(Function &, FunctionAnalysisManager &)的类声明。 我认为您的重新实现应该是 15-20 行代码。

如果您之前没有写过通行证，我会概述一下。 它需要遍历基本块并查看每个块的终止符是否为 <ReturnInst>()，并将那些添加到列表中。 迭代后，如果列表中的返回少于两个，则提前返回。

否则你会创建一个新的基本块。 如果返回类型不是 void，则需要创建一个phi 节点并使用返回的传入值填充它，然后创建一个返回 phi 节点的返回。 如果它无效，您将返回一个新的无效返回值。 最后，您将所有旧返回替换为新块的分支。 完毕。