llvm/llvm/unittests/Transforms/Coroutines/ExtraRematTest.cpp

//===- ExtraRematTest.cpp - Coroutines unit tests -------------------------===//
//
// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
// See https://llvm.org/LICENSE.txt for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
//
//===----------------------------------------------------------------------===//

#include "llvm/AsmParser/Parser.h"
#include "llvm/IR/Module.h"
#include "llvm/Passes/PassBuilder.h"
#include "llvm/Support/SourceMgr.h"
#include "llvm/Testing/Support/Error.h"
#include "llvm/Transforms/Coroutines/ABI.h"
#include "llvm/Transforms/Coroutines/CoroSplit.h"
#include "gtest/gtest.h"

usingnamespacellvm;

namespace {

struct ExtraRematTest : public testing::Test { … };

StringRef Text = …;

// Materializable callback with extra rematerialization
bool ExtraMaterializable(Instruction &I) { … }

TEST_F(ExtraRematTest, TestCoroRematDefault) { … }

TEST_F(ExtraRematTest, TestCoroRematWithCallback) { … }

StringRef TextCoroBeginCustomABI = …;

// SwitchABI with overridden isMaterializable
class ExtraCustomABI : public coro::SwitchABI { … };

TEST_F(ExtraRematTest, TestCoroRematWithCustomABI) { … }

} // namespace