mirror of
https://github.com/llvm/llvm-project.git
synced 2025-05-14 11:36:09 +00:00
c6cf329502
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
43 lines
1.3 KiB
C++
43 lines
1.3 KiB
C++
//===-- LinkInModulesPass.h - 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
|
|
///
|
|
/// This file provides a pass to link in Modules from a provided
|
|
/// BackendConsumer.
|
|
///
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_BITCODE_LINKINMODULESPASS_H
|
|
#define LLVM_BITCODE_LINKINMODULESPASS_H
|
|
|
|
#include "BackendConsumer.h"
|
|
#include "llvm/IR/PassManager.h"
|
|
|
|
namespace llvm {
|
|
class Module;
|
|
class ModulePass;
|
|
class Pass;
|
|
|
|
/// Create and return a pass that links in Moduels from a provided
|
|
/// BackendConsumer to a given primary Module. Note that this pass is designed
|
|
/// for use with the legacy pass manager.
|
|
class LinkInModulesPass : public PassInfoMixin<LinkInModulesPass> {
|
|
clang::BackendConsumer *BC;
|
|
bool ShouldLinkFiles;
|
|
|
|
public:
|
|
LinkInModulesPass(clang::BackendConsumer *BC, bool ShouldLinkFiles = true);
|
|
|
|
PreservedAnalyses run(Module &M, AnalysisManager<Module> &);
|
|
static bool isRequired() { return true; }
|
|
};
|
|
|
|
} // namespace llvm
|
|
|
|
#endif
|