mirror of
https://github.com/llvm/llvm-project.git
synced 2025-04-25 21:26:07 +00:00
a72d7eea54
This patch improves the design of the IncrementalParser and Interpreter classes. Now the incremental parser is only responsible for building the partial translation unit declaration and the AST, while the Interpreter fills in the lower level llvm::Module and other JIT-related infrastructure. Finally the Interpreter class now orchestrates the AST and the LLVM IR with the IncrementalParser and IncrementalExecutor classes. The design improvement allows us to rework some of the logic that extracts an interpreter value into the clang::Value object. The new implementation simplifies use-cases which are used for out-of-process execution by allowing interpreter to be inherited or customized with an clang::ASTConsumer. This change will enable completing the pretty printing work which is in llvm/llvm-project#84769
59 lines
1.9 KiB
C++
59 lines
1.9 KiB
C++
//===----------- DeviceOffload.h - Device Offloading ------------*- 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file implements classes required for offloading to CUDA devices.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef LLVM_CLANG_LIB_INTERPRETER_DEVICE_OFFLOAD_H
|
|
#define LLVM_CLANG_LIB_INTERPRETER_DEVICE_OFFLOAD_H
|
|
|
|
#include "IncrementalParser.h"
|
|
#include "llvm/Support/FileSystem.h"
|
|
#include "llvm/Support/VirtualFileSystem.h"
|
|
|
|
namespace clang {
|
|
struct PartialTranslationUnit;
|
|
class CompilerInstance;
|
|
class CodeGenOptions;
|
|
class TargetOptions;
|
|
|
|
class IncrementalCUDADeviceParser : public IncrementalParser {
|
|
const std::list<PartialTranslationUnit> &PTUs;
|
|
|
|
public:
|
|
IncrementalCUDADeviceParser(
|
|
std::unique_ptr<CompilerInstance> DeviceInstance,
|
|
CompilerInstance &HostInstance,
|
|
llvm::IntrusiveRefCntPtr<llvm::vfs::InMemoryFileSystem> VFS,
|
|
llvm::Error &Err, const std::list<PartialTranslationUnit> &PTUs);
|
|
|
|
llvm::Expected<TranslationUnitDecl *> Parse(llvm::StringRef Input) override;
|
|
|
|
// Generate PTX for the last PTU.
|
|
llvm::Expected<llvm::StringRef> GeneratePTX();
|
|
|
|
// Generate fatbinary contents in memory
|
|
llvm::Error GenerateFatbinary();
|
|
|
|
~IncrementalCUDADeviceParser();
|
|
|
|
protected:
|
|
std::unique_ptr<CompilerInstance> DeviceCI;
|
|
int SMVersion;
|
|
llvm::SmallString<1024> PTXCode;
|
|
llvm::SmallVector<char, 1024> FatbinContent;
|
|
llvm::IntrusiveRefCntPtr<llvm::vfs::InMemoryFileSystem> VFS;
|
|
CodeGenOptions &CodeGenOpts; // Intentionally a reference.
|
|
const TargetOptions &TargetOpts;
|
|
};
|
|
|
|
} // namespace clang
|
|
|
|
#endif // LLVM_CLANG_LIB_INTERPRETER_DEVICE_OFFLOAD_H
|