================ @@ -0,0 +1,62 @@ +//===- OffloadWrapper.h --r-------------------------------------*- 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 +// +//===----------------------------------------------------------------------===// + +#ifndef LLVM_FRONTEND_OFFLOADING_OFFLOADWRAPPER_H +#define LLVM_FRONTEND_OFFLOADING_OFFLOADWRAPPER_H + +#include "llvm/ADT/ArrayRef.h" +#include "llvm/IR/Module.h" + +namespace llvm { +namespace offloading { +/// Class for embedding and registering offloading images and related objects in +/// a Module. +class OffloadWrapper { +public: + using EntryArrayTy = std::pair<GlobalVariable *, GlobalVariable *>; + + OffloadWrapper(const Twine &Suffix = "", bool EmitSurfacesAndTextures = true) + : Suffix(Suffix.str()), EmitSurfacesAndTextures(EmitSurfacesAndTextures) { + } + + /// Wraps the input device images into the module \p M as global symbols and + /// registers the images with the OpenMP Offloading runtime libomptarget. + /// \param EntryArray Optional pair pointing to the `__start` and `__stop` + /// symbols holding the `__tgt_offload_entry` array. + llvm::Error wrapOpenMPBinaries( + llvm::Module &M, llvm::ArrayRef<llvm::ArrayRef<char>> Images, + std::optional<EntryArrayTy> EntryArray = std::nullopt) const; + + /// Wraps the input fatbinary image into the module \p M as global symbols and + /// registers the images with the CUDA runtime. + /// \param EntryArray Optional pair pointing to the `__start` and `__stop` + /// symbols holding the `__tgt_offload_entry` array. + llvm::Error + wrapCudaBinary(llvm::Module &M, llvm::ArrayRef<char> Images, + std::optional<EntryArrayTy> EntryArray = std::nullopt) const; + + /// Wraps the input bundled image into the module \p M as global symbols and + /// registers the images with the HIP runtime. + /// \param EntryArray Optional pair pointing to the `__start` and `__stop` + /// symbols holding the `__tgt_offload_entry` array. + llvm::Error + wrapHIPBinary(llvm::Module &M, llvm::ArrayRef<char> Images, + std::optional<EntryArrayTy> EntryArray = std::nullopt) const; + +protected: + /// Suffix used when emitting symbols. It defaults to the empty string. + std::string Suffix; + + /// Whether to emit surface and textures registration code. It defaults to + /// false. + bool EmitSurfacesAndTextures; ---------------- jhuber6 wrote:
So, I wasn't sure about this either. I know that CUDA emits these `__cudaRegisterSurface` calls, but I can't seem to find them in any of the exported libraries. It caused linker errors due to that and I was too lazy to fix it. Wondering if they've been deprecated, maybe @tra knows. https://github.com/llvm/llvm-project/pull/78057 _______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
