From 94936248fd06eceb2076041b1827c2391458fd02 Mon Sep 17 00:00:00 2001 From: Sergio Afonso Date: Tue, 8 Jul 2025 12:27:08 +0100 Subject: [PATCH] [OpenMP][OMPIRBuilder] Add device shared memory allocation support This patch adds the `__kmpc_alloc_shared` and `__kmpc_free_shared` DeviceRTL functions to the list of those the OMPIRBuilder is able to create. --- .../llvm/Frontend/OpenMP/OMPIRBuilder.h | 23 ++++++++++++++++ llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp | 27 +++++++++++++++++++ 2 files changed, 50 insertions(+) diff --git a/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h b/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h index f70659120e1e6..0fb664aa5f888 100644 --- a/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h +++ b/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h @@ -2907,6 +2907,29 @@ class OpenMPIRBuilder { LLVM_ABI CallInst *createOMPFree(const LocationDescription &Loc, Value *Addr, Value *Allocator, std::string Name = ""); + /// Create a runtime call for kmpc_alloc_shared. + /// + /// \param Loc The insert and source location description. + /// \param VarType Type of variable to be allocated. + /// \param Name Name of call Instruction. + /// + /// \returns CallInst to the kmpc_alloc_shared call. + LLVM_ABI CallInst *createOMPAllocShared(const LocationDescription &Loc, + Type *VarType, + const Twine &Name = Twine("")); + + /// Create a runtime call for kmpc_free_shared. + /// + /// \param Loc The insert and source location description. + /// \param Addr Value obtained from the corresponding kmpc_alloc_shared call. + /// \param VarType Type of variable to be freed. + /// \param Name Name of call Instruction. + /// + /// \returns CallInst to the kmpc_free_shared call. + LLVM_ABI CallInst *createOMPFreeShared(const LocationDescription &Loc, + Value *Addr, Type *VarType, + const Twine &Name = Twine("")); + /// Create a runtime call for kmpc_threadprivate_cached /// /// \param Loc The insert and source location description. diff --git a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp index c16b0dde1a3da..75af18518ff84 100644 --- a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp +++ b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp @@ -6631,6 +6631,33 @@ CallInst *OpenMPIRBuilder::createOMPFree(const LocationDescription &Loc, return Builder.CreateCall(Fn, Args, Name); } +CallInst *OpenMPIRBuilder::createOMPAllocShared(const LocationDescription &Loc, + Type *VarType, + const Twine &Name) { + IRBuilder<>::InsertPointGuard IPG(Builder); + updateToLocation(Loc); + + const DataLayout &DL = M.getDataLayout(); + Value *Args[] = {Builder.getInt64(DL.getTypeStoreSize(VarType))}; + Function *Fn = getOrCreateRuntimeFunctionPtr(OMPRTL___kmpc_alloc_shared); + CallInst *Call = Builder.CreateCall(Fn, Args, Name); + Call->addRetAttr( + Attribute::getWithAlignment(M.getContext(), DL.getPrefTypeAlign(Int64))); + return Call; +} + +CallInst *OpenMPIRBuilder::createOMPFreeShared(const LocationDescription &Loc, + Value *Addr, Type *VarType, + const Twine &Name) { + IRBuilder<>::InsertPointGuard IPG(Builder); + updateToLocation(Loc); + + Value *Args[] = { + Addr, Builder.getInt64(M.getDataLayout().getTypeStoreSize(VarType))}; + Function *Fn = getOrCreateRuntimeFunctionPtr(OMPRTL___kmpc_free_shared); + return Builder.CreateCall(Fn, Args, Name); +} + CallInst *OpenMPIRBuilder::createOMPInteropInit( const LocationDescription &Loc, Value *InteropVar, omp::OMPInteropType InteropType, Value *Device, Value *NumDependences,