================ @@ -568,32 +590,45 @@ void createRegisterFatbinFunction(Module &M, GlobalVariable *FatbinDesc, } // namespace -Error wrapOpenMPBinaries(Module &M, ArrayRef<ArrayRef<char>> Images) { - GlobalVariable *Desc = createBinDesc(M, Images); +Error OffloadWrapper::wrapOpenMPBinaries( + Module &M, ArrayRef<ArrayRef<char>> Images, + std::optional<EntryArrayTy> EntryArray) const { + GlobalVariable *Desc = createBinDesc( + M, Images, + EntryArray + ? *EntryArray + : offloading::getOffloadEntryArray(M, "omp_offloading_entries"), + Suffix); if (!Desc) return createStringError(inconvertibleErrorCode(), "No binary descriptors created."); - createRegisterFunction(M, Desc); - createUnregisterFunction(M, Desc); + createRegisterFunction(M, Desc, Suffix); + createUnregisterFunction(M, Desc, Suffix); return Error::success(); } -Error wrapCudaBinary(Module &M, ArrayRef<char> Image) { - GlobalVariable *Desc = createFatbinDesc(M, Image, /* IsHIP */ false); +Error OffloadWrapper::wrapCudaBinary( + Module &M, ArrayRef<char> Image, + std::optional<EntryArrayTy> EntryArray) const { + GlobalVariable *Desc = createFatbinDesc(M, Image, /* IsHIP */ false, Suffix); if (!Desc) return createStringError(inconvertibleErrorCode(), "No fatinbary section created."); - createRegisterFatbinFunction(M, Desc, /* IsHIP */ false); + createRegisterFatbinFunction(M, Desc, /* IsHIP */ false, EntryArray, Suffix, + EmitSurfacesAndTextures); return Error::success(); } -Error wrapHIPBinary(Module &M, ArrayRef<char> Image) { - GlobalVariable *Desc = createFatbinDesc(M, Image, /* IsHIP */ true); +Error OffloadWrapper::wrapHIPBinary( + Module &M, ArrayRef<char> Image, + std::optional<EntryArrayTy> EntryArray) const { + GlobalVariable *Desc = createFatbinDesc(M, Image, /* IsHIP */ true, Suffix); if (!Desc) return createStringError(inconvertibleErrorCode(), "No fatinbary section created."); - createRegisterFatbinFunction(M, Desc, /* IsHIP */ true); + createRegisterFatbinFunction(M, Desc, /* IsHIP */ true, EntryArray, Suffix, ---------------- jhuber6 wrote:
Can you fix these comments while you're at it? LLVM inline comments should be `/*IsHIP=*/` 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