TheRock/compiler/amd-llvm/clang/test/CodeGenCoroutines/pr59181.cpp

// Test for PR59181. Tests that no conditional cleanup is created around await_suspend.
//
// REQUIRES: x86-registered-target
//
// RUN: %clang_cc1 -triple x86_64-unknown-linux-gnu -emit-llvm %s -o - -std=c++20 -disable-llvm-passes -fsanitize-address-use-after-scope | FileCheck %s

#include "Inputs/coroutine.h"

struct Task {
  int value_;
  struct promise_type {
    Task get_return_object() {
      return Task{0};
    }

    std::suspend_never initial_suspend() noexcept {
      return {};
    }

    std::suspend_never final_suspend() noexcept {
      return {};
    }

    void return_value(Task t) noexcept {}
    void unhandled_exception() noexcept {}

    auto await_transform(Task t) {
      struct Suspension {
        auto await_ready() noexcept { return false;}
        auto await_suspend(std::coroutine_handle<> coro) {
          coro.destroy();
        }

        auto await_resume() noexcept {
          return 0;
        }
      };
      return Suspension{};
    }
  };
};

Task bar(bool cond) {
  co_return cond ? Task{ co_await Task{}}: Task{};
}

void foo() {
  bar(true);
}

// CHECK: cleanup.cont:{{.*}}
// CHECK-NEXT: load i8
// CHECK-NEXT: icmp ne
// CHECK-NEXT: store i1 false

// CHECK: await.suspend:{{.*}}
// CHECK-NOT: call void @llvm.lifetime
// CHECK: call void @llvm.coro.await.suspend.void(
// CHECK-NEXT: %{{[0-9]+}} = call i8 @llvm.coro.suspend(