2ab4c2880d
Read discussion : https://github.com/llvm/llvm-project/pull/136404#discussion_r2059149768 and the following comments for context Motivation 1) `IncrementalAction` is designed to keep Frontend statealive across inputs. As per the docstring: “IncrementalAction ensures it keeps its underlying action's objects alive as long as the IncrementalParser needs them.” 2) To align responsibilities with that contract, the parser layer (host: `IncrementalParser`, device: `IncrementalCUDADeviceParser`) should manage PTU registration and module generation, while the interpreter orchestrates at a higher level. What this PR does 1) Moves CodeGen surfaces behind IncrementalAction: GenModule(), getCodeGen(), and the cached “first CodeGen module” now live in IncrementalAction. 2) Moves PTU ownership to the parser layer: Adds IncrementalParser::RegisterPTU(…) (and device counterpart) 3) Add device-side registration in IncrementalCUDADeviceParser. 4) Remove Interpreter::{getCodeGen, GenModule, RegisterPTU}.
56 lines
1.7 KiB
C++
56 lines
1.7 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 IncrementalAction;
|
|
|
|
class IncrementalCUDADeviceParser : public IncrementalParser {
|
|
|
|
public:
|
|
IncrementalCUDADeviceParser(
|
|
CompilerInstance &DeviceInstance, CompilerInstance &HostInstance,
|
|
IncrementalAction *DeviceAct,
|
|
llvm::IntrusiveRefCntPtr<llvm::vfs::InMemoryFileSystem> VFS,
|
|
llvm::Error &Err, std::list<PartialTranslationUnit> &PTUs);
|
|
|
|
// Generate PTX for the last PTU.
|
|
llvm::Expected<llvm::StringRef> GeneratePTX();
|
|
|
|
// Generate fatbinary contents in memory
|
|
llvm::Error GenerateFatbinary();
|
|
|
|
~IncrementalCUDADeviceParser();
|
|
|
|
protected:
|
|
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
|