Files
carbon-lang/toolchain/codegen/codegen.cpp
T
2025-03-07 19:34:31 +00:00

65 lines
2.0 KiB
C++

// Part of the Carbon Language project, under the Apache License v2.0 with LLVM
// Exceptions. See /LICENSE for license information.
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
#include "toolchain/codegen/codegen.h"
#include <memory>
#include "llvm/IR/LegacyPassManager.h"
#include "llvm/MC/TargetRegistry.h"
#include "llvm/Target/TargetOptions.h"
#include "llvm/TargetParser/Host.h"
namespace Carbon {
auto CodeGen::Make(llvm::Module& module, llvm::StringRef target_triple,
llvm::raw_pwrite_stream& errors) -> std::optional<CodeGen> {
std::string error;
const llvm::Target* target =
llvm::TargetRegistry::lookupTarget(target_triple, error);
if (!target) {
errors << "error: invalid target: " << error << "\n";
return {};
}
module.setTargetTriple(llvm::Triple(target_triple));
constexpr llvm::StringLiteral CPU = "generic";
constexpr llvm::StringLiteral Features = "";
llvm::TargetOptions target_opts;
CodeGen codegen(module, errors);
codegen.target_machine_.reset(target->createTargetMachine(
target_triple, CPU, Features, target_opts, llvm::Reloc::PIC_));
return codegen;
}
auto CodeGen::EmitAssembly(llvm::raw_pwrite_stream& out) -> bool {
return EmitCode(out, llvm::CodeGenFileType::AssemblyFile);
}
auto CodeGen::EmitObject(llvm::raw_pwrite_stream& out) -> bool {
return EmitCode(out, llvm::CodeGenFileType::ObjectFile);
}
auto CodeGen::EmitCode(llvm::raw_pwrite_stream& out,
llvm::CodeGenFileType file_type) -> bool {
module_.setDataLayout(target_machine_->createDataLayout());
// Using the legacy PM to generate the assembly since the new PM
// does not work with this yet.
// TODO: Make the new PM work with the codegen pipeline.
llvm::legacy::PassManager pass;
// Note that this returns true on an error.
if (target_machine_->addPassesToEmitFile(pass, out, nullptr, file_type)) {
errors_ << "error: unable to emit to this file\n";
return false;
}
pass.run(module_);
return true;
}
} // namespace Carbon