如何正确修改llvm pass中的IR?

问题描述 投票:0回答:1

我正在研究 LLVM pass,其目标是将用户定义的函数转换为 coroutines。例如,一个简单的 noop 函数:

void f() {}

转变为:

define ptr @f_coro() {
init:
  %promise = alloca %CoroPromise, align 8
  call void @init_promise(ptr %promise) ; my own API
  %id = call token @llvm.coro.id(i32 0, ptr %promise, ptr null, ptr null)
  %size = call i32 @llvm.coro.size.i32()
  %hdl_alloc = call ptr @malloc(i32 %size)
  %hdl = call ptr @llvm.coro.begin(token %id, ptr %hdl_alloc)
  %0 = call i8 @llvm.coro.suspend(token none, i1 false)
  switch i8 %0, label %suspend [
    i8 0, label %terminator.1
    i8 1, label %cleanup
  ]

cleanup:                                       
  %1 = call ptr @llvm.coro.free(token %id, ptr %hdl)
  call void @free(ptr %1)
  br label %suspend

suspend:                                         
  %2 = call i1 @llvm.coro.end(ptr %hdl, i1 false, token none)
  ret ptr %hdl

terminator.1:                                   
  call void @set_ret_val(ptr %promise, i32 0) ; my own API
  %3 = call i8 @llvm.coro.suspend(token none, i1 false)
  switch i8 %3, label %suspend [
    i8 0, label %cleanup
    i8 1, label %cleanup
  ]
}

但是,当我使用 clang(版本 18.0.0)运行通行证时:

clang++ -fno-discard-value-names -std=c++2a -emit-llvm -S -fpass-plugin=Pass.so sample.cpp

它省略了我生成的一些指令(具体来说,@llvm.coro.id),这导致后续代码编译失败。我应该如何正确执行此类操作?”

我尝试对预先存在的字节码使用 opt 而不是 clang。然而,这使管道变得复杂。

clang llvm coroutine
1个回答
0
投票

忘记为我的函数设置

presplitcoroutine
属性。

© www.soinside.com 2019 - 2024. All rights reserved.