mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-15 06:56:08 +00:00

In this patch, we create a new ModulePass that mimics the LinkInModules API from CodeGenAction.cpp, and a new command line option to enable the pass. As part of the implementation, we needed to refactor the BackendConsumer class definition into a new separate header (instead of embedded in CodeGenAction.cpp). With this new pass, we can now re-link bitcodes supplied via the -mlink-built-in bitcodes as part of the RunOptimizationPipeline. With the re-linking pass, we now handle cases where new device library functions are introduced as part of the optimization pipeline. Previously, these newly introduced functions (for example a fused sincos call) would result in a linking error due to a missing function definition. This new pass can be initiated via: -mllvm -relink-builtin-bitcode-postop Also note we intentionally exclude bitcodes supplied via the -mlink-bitcode-file option from the second linking step
30 lines
1017 B
C++
30 lines
1017 B
C++
//===-- LinkInModulesPass.cpp - Module Linking pass --------------- C++ -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
/// \file
|
|
///
|
|
/// LinkInModulesPass implementation.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#include "LinkInModulesPass.h"
|
|
#include "BackendConsumer.h"
|
|
|
|
using namespace llvm;
|
|
|
|
LinkInModulesPass::LinkInModulesPass(clang::BackendConsumer *BC,
|
|
bool ShouldLinkFiles)
|
|
: BC(BC), ShouldLinkFiles(ShouldLinkFiles) {}
|
|
|
|
PreservedAnalyses LinkInModulesPass::run(Module &M, ModuleAnalysisManager &AM) {
|
|
|
|
if (BC && BC->LinkInModules(&M, ShouldLinkFiles))
|
|
report_fatal_error("Bitcode module linking failed, compilation aborted!");
|
|
|
|
return PreservedAnalyses::all();
|
|
}
|