1*06c3fb27SDimitry Andric //===----------- DeviceOffload.h - Device Offloading ------------*- C++ -*-===// 2*06c3fb27SDimitry Andric // 3*06c3fb27SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. 4*06c3fb27SDimitry Andric // See https://llvm.org/LICENSE.txt for license information. 5*06c3fb27SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception 6*06c3fb27SDimitry Andric // 7*06c3fb27SDimitry Andric //===----------------------------------------------------------------------===// 8*06c3fb27SDimitry Andric // 9*06c3fb27SDimitry Andric // This file implements classes required for offloading to CUDA devices. 10*06c3fb27SDimitry Andric // 11*06c3fb27SDimitry Andric //===----------------------------------------------------------------------===// 12*06c3fb27SDimitry Andric 13*06c3fb27SDimitry Andric #ifndef LLVM_CLANG_LIB_INTERPRETER_DEVICE_OFFLOAD_H 14*06c3fb27SDimitry Andric #define LLVM_CLANG_LIB_INTERPRETER_DEVICE_OFFLOAD_H 15*06c3fb27SDimitry Andric 16*06c3fb27SDimitry Andric #include "IncrementalParser.h" 17*06c3fb27SDimitry Andric #include "llvm/Support/FileSystem.h" 18*06c3fb27SDimitry Andric #include "llvm/Support/VirtualFileSystem.h" 19*06c3fb27SDimitry Andric 20*06c3fb27SDimitry Andric namespace clang { 21*06c3fb27SDimitry Andric 22*06c3fb27SDimitry Andric class IncrementalCUDADeviceParser : public IncrementalParser { 23*06c3fb27SDimitry Andric public: 24*06c3fb27SDimitry Andric IncrementalCUDADeviceParser( 25*06c3fb27SDimitry Andric Interpreter &Interp, std::unique_ptr<CompilerInstance> Instance, 26*06c3fb27SDimitry Andric IncrementalParser &HostParser, llvm::LLVMContext &LLVMCtx, 27*06c3fb27SDimitry Andric llvm::IntrusiveRefCntPtr<llvm::vfs::InMemoryFileSystem> VFS, 28*06c3fb27SDimitry Andric llvm::Error &Err); 29*06c3fb27SDimitry Andric 30*06c3fb27SDimitry Andric llvm::Expected<PartialTranslationUnit &> 31*06c3fb27SDimitry Andric Parse(llvm::StringRef Input) override; 32*06c3fb27SDimitry Andric 33*06c3fb27SDimitry Andric // Generate PTX for the last PTU 34*06c3fb27SDimitry Andric llvm::Expected<llvm::StringRef> GeneratePTX(); 35*06c3fb27SDimitry Andric 36*06c3fb27SDimitry Andric // Generate fatbinary contents in memory 37*06c3fb27SDimitry Andric llvm::Error GenerateFatbinary(); 38*06c3fb27SDimitry Andric 39*06c3fb27SDimitry Andric ~IncrementalCUDADeviceParser(); 40*06c3fb27SDimitry Andric 41*06c3fb27SDimitry Andric protected: 42*06c3fb27SDimitry Andric IncrementalParser &HostParser; 43*06c3fb27SDimitry Andric int SMVersion; 44*06c3fb27SDimitry Andric llvm::SmallString<1024> PTXCode; 45*06c3fb27SDimitry Andric llvm::SmallVector<char, 1024> FatbinContent; 46*06c3fb27SDimitry Andric llvm::IntrusiveRefCntPtr<llvm::vfs::InMemoryFileSystem> VFS; 47*06c3fb27SDimitry Andric }; 48*06c3fb27SDimitry Andric 49*06c3fb27SDimitry Andric } // namespace clang 50*06c3fb27SDimitry Andric 51*06c3fb27SDimitry Andric #endif // LLVM_CLANG_LIB_INTERPRETER_DEVICE_OFFLOAD_H 52