[Mlir-commits] [llvm] [mlir] [mlir][llvm][OpenMP] Support iterator modifier in depend clause (PR #190026)
llvmlistbot at llvm.org
llvmlistbot at llvm.org
Wed Apr 1 12:48:26 PDT 2026
llvmbot wrote:
<!--LLVM PR SUMMARY COMMENT-->
@llvm/pr-subscribers-mlir-llvm
@llvm/pr-subscribers-mlir
Author: Chi-Chun, Chen (chichunchen)
<details>
<summary>Changes</summary>
Add translation from the MLIR OpenMP depend clause with iterator modifier to LLVM IR. `buildDependData` (in OpenMPToLLVMIRTranslation) allocates a single `kmp_dep_info` array sized to hold both locator (non-iterated) and iterated entries. Locator dependencies use the existing static path (a vector of `DependData`), while iterated dependencies use a dynamically-sized path (`DepArray`, `NumDeps`).
The reason both paths are not unified under the dynamic allocation is that the existing locator path emits actual `kmp_dep_info` entries inside OMPIRBuilder methods (`createTask`, `createTarget`), whereas the iterator path must emit the iterator loop in OpenMPToLLVMIRTranslation (since the convention is to not pass MLIR ops into the OMPIRBuilder). Unifying them would require modifying existing depend clause tests.
The `OMPIRBuilder::DependenciesInfo` struct is extended to hold either a `SmallVector<DependData>` (locator path) or a pre-built `{DepArray, NumDeps}` pair (iterator path). The single-entry `emitTaskDependency` helper is made public so the translation layer can fill individual `kmp_dep_info` entries inside the iterator loop body.
This patch is part of the feature work for #<!-- -->188061.
Assisted with copilot.
---
Patch is 38.88 KiB, truncated to 20.00 KiB below, full version: https://github.com/llvm/llvm-project/pull/190026.diff
6 Files Affected:
- (modified) llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h (+32-11)
- (modified) llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp (+70-52)
- (modified) llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp (+2-1)
- (modified) mlir/lib/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.cpp (+132-39)
- (modified) mlir/test/Target/LLVMIR/openmp-iterator.mlir (+216-1)
- (modified) mlir/test/Target/LLVMIR/openmp-todo.mlir (-30)
``````````diff
diff --git a/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h b/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h
index 383fd9d94661a..bbbb0084859f7 100644
--- a/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h
+++ b/llvm/include/llvm/Frontend/OpenMP/OMPIRBuilder.h
@@ -1511,6 +1511,28 @@ class OpenMPIRBuilder {
: DepKind(DepKind), DepValueType(DepValueType), DepVal(DepVal) {}
};
+ /// A struct to pack static and dynamic dependency information for a task.
+ ///
+ /// For fixed-count (non-iterator) dependencies, callers populate \p Deps
+ /// and the builder allocates and fills the kmp_dep_info array internally.
+ /// For iterator-based dependencies, the caller pre-builds the array and
+ /// sets \p NumDeps and \p DepArray directly.
+ struct DependenciesInfo {
+ SmallVector<DependData> Deps; // vector of dependencies
+ Value *NumDeps; // number of kmp_dep_info entries (used by iterator path)
+ Value *DepArray; // kmp_dep_info array (used by iterator path)
+
+ DependenciesInfo() : Deps(), NumDeps(nullptr), DepArray(nullptr) {}
+ DependenciesInfo(SmallVector<DependData> D)
+ : Deps(std::move(D)), NumDeps(nullptr), DepArray(nullptr) {}
+
+ bool empty() const { return Deps.empty() && DepArray == nullptr; }
+ };
+
+ /// Store one kmp_dep_info entry at the given \p Entry pointer.
+ LLVM_ABI void emitTaskDependency(IRBuilderBase &Builder, Value *Entry,
+ const DependData &Dep);
+
/// Return the LLVM struct type matching runtime `kmp_task_affinity_info_t`.
/// `{ kmp_intptr_t base_addr; size_t len; flags (bitfield storage as i32) }`
LLVM_ABI llvm::StructType *getKmpTaskAffinityInfoTy();
@@ -1579,8 +1601,8 @@ class OpenMPIRBuilder {
/// cannot be resumed until execution of the structured
/// block that is associated with the generated task is
/// completed.
- /// \param Dependencies Vector of DependData objects holding information of
- /// dependencies as specified by the 'depend' clause.
+ /// \param Dependencies Dependencies info holding either a vector of
+ /// DependData objects or a pre-built dependency array.
/// \param Affinities AffinityData object holding information of accumulated
/// affinities as specified by the 'affinity' clause.
/// \param EventHandle If present, signifies the event handle as part of
@@ -1591,7 +1613,7 @@ class OpenMPIRBuilder {
LLVM_ABI InsertPointOrErrorTy createTask(
const LocationDescription &Loc, InsertPointTy AllocaIP,
BodyGenCallbackTy BodyGenCB, bool Tied = true, Value *Final = nullptr,
- Value *IfCondition = nullptr, SmallVector<DependData> Dependencies = {},
+ Value *IfCondition = nullptr, DependenciesInfo Dependencies = {},
AffinityData Affinities = {}, bool Mergeable = false,
Value *EventHandle = nullptr, Value *Priority = nullptr);
@@ -2888,15 +2910,14 @@ class OpenMPIRBuilder {
/// \param DeviceID Identifier for the device via the 'device' clause.
/// \param RTLoc Source location identifier
/// \param AllocaIP The insertion point to be used for alloca instructions.
- /// \param Dependencies Vector of DependData objects holding information of
- /// dependencies as specified by the 'depend' clause.
+ /// \param Dependencies Dependencies info as specified by the 'depend' clause.
/// \param HasNoWait True if the target construct had 'nowait' on it, false
/// otherwise
- LLVM_ABI InsertPointOrErrorTy emitTargetTask(
- TargetTaskBodyCallbackTy TaskBodyCB, Value *DeviceID, Value *RTLoc,
- OpenMPIRBuilder::InsertPointTy AllocaIP,
- const SmallVector<llvm::OpenMPIRBuilder::DependData> &Dependencies,
- const TargetDataRTArgs &RTArgs, bool HasNoWait);
+ LLVM_ABI InsertPointOrErrorTy
+ emitTargetTask(TargetTaskBodyCallbackTy TaskBodyCB, Value *DeviceID,
+ Value *RTLoc, OpenMPIRBuilder::InsertPointTy AllocaIP,
+ const DependenciesInfo &Dependencies,
+ const TargetDataRTArgs &RTArgs, bool HasNoWait);
/// Emit the arguments to be passed to the runtime library based on the
/// arrays of base pointers, pointers, sizes, map types, and mappers. If
@@ -3537,7 +3558,7 @@ class OpenMPIRBuilder {
TargetBodyGenCallbackTy BodyGenCB,
TargetGenArgAccessorsCallbackTy ArgAccessorFuncCB,
CustomMapperCallbackTy CustomMapperCB,
- const SmallVector<DependData> &Dependencies, bool HasNowait = false,
+ const DependenciesInfo &Dependencies = {}, bool HasNowait = false,
Value *DynCGroupMem = nullptr,
omp::OMPDynGroupprivateFallbackType DynCGroupMemFallback =
omp::OMPDynGroupprivateFallbackType::Abort);
diff --git a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
index ae8f261ff98b2..fbf26b201c3b1 100644
--- a/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
+++ b/llvm/lib/Frontend/OpenMP/OMPIRBuilder.cpp
@@ -1965,6 +1965,28 @@ void OpenMPIRBuilder::createTaskyield(const LocationDescription &Loc) {
emitTaskyieldImpl(Loc);
}
+void OpenMPIRBuilder::emitTaskDependency(IRBuilderBase &Builder, Value *Entry,
+ const DependData &Dep) {
+ // Store the pointer to the variable
+ Value *Addr = Builder.CreateStructGEP(
+ DependInfo, Entry,
+ static_cast<unsigned int>(RTLDependInfoFields::BaseAddr));
+ Value *DepValPtr = Builder.CreatePtrToInt(Dep.DepVal, Builder.getInt64Ty());
+ Builder.CreateStore(DepValPtr, Addr);
+ // Store the size of the variable
+ Value *Size = Builder.CreateStructGEP(
+ DependInfo, Entry, static_cast<unsigned int>(RTLDependInfoFields::Len));
+ Builder.CreateStore(
+ Builder.getInt64(M.getDataLayout().getTypeStoreSize(Dep.DepValueType)),
+ Size);
+ // Store the dependency kind
+ Value *Flags = Builder.CreateStructGEP(
+ DependInfo, Entry, static_cast<unsigned int>(RTLDependInfoFields::Flags));
+ Builder.CreateStore(ConstantInt::get(Builder.getInt8Ty(),
+ static_cast<unsigned int>(Dep.DepKind)),
+ Flags);
+}
+
// Processes the dependencies in Dependencies and does the following
// - Allocates space on the stack of an array of DependInfo objects
// - Populates each DependInfo object with relevant information of
@@ -1995,7 +2017,6 @@ static Value *emitTaskDependencies(
IRBuilderBase &Builder = OMPBuilder.Builder;
Type *DependInfo = OMPBuilder.DependInfo;
- Module &M = OMPBuilder.M;
Value *DepArray = nullptr;
OpenMPIRBuilder::InsertPointTy OldIP = Builder.saveIP();
@@ -2010,26 +2031,7 @@ static Value *emitTaskDependencies(
for (const auto &[DepIdx, Dep] : enumerate(Dependencies)) {
Value *Base =
Builder.CreateConstInBoundsGEP2_64(DepArrayTy, DepArray, 0, DepIdx);
- // Store the pointer to the variable
- Value *Addr = Builder.CreateStructGEP(
- DependInfo, Base,
- static_cast<unsigned int>(RTLDependInfoFields::BaseAddr));
- Value *DepValPtr = Builder.CreatePtrToInt(Dep.DepVal, Builder.getInt64Ty());
- Builder.CreateStore(DepValPtr, Addr);
- // Store the size of the variable
- Value *Size = Builder.CreateStructGEP(
- DependInfo, Base, static_cast<unsigned int>(RTLDependInfoFields::Len));
- Builder.CreateStore(
- Builder.getInt64(M.getDataLayout().getTypeStoreSize(Dep.DepValueType)),
- Size);
- // Store the dependency kind
- Value *Flags = Builder.CreateStructGEP(
- DependInfo, Base,
- static_cast<unsigned int>(RTLDependInfoFields::Flags));
- Builder.CreateStore(
- ConstantInt::get(Builder.getInt8Ty(),
- static_cast<unsigned int>(Dep.DepKind)),
- Flags);
+ OMPBuilder.emitTaskDependency(Builder, Base, Dep);
}
return DepArray;
}
@@ -2441,8 +2443,8 @@ llvm::StructType *OpenMPIRBuilder::getKmpTaskAffinityInfoTy() {
OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::createTask(
const LocationDescription &Loc, InsertPointTy AllocaIP,
BodyGenCallbackTy BodyGenCB, bool Tied, Value *Final, Value *IfCondition,
- SmallVector<DependData> Dependencies, AffinityData Affinities,
- bool Mergeable, Value *EventHandle, Value *Priority) {
+ DependenciesInfo Dependencies, AffinityData Affinities, bool Mergeable,
+ Value *EventHandle, Value *Priority) {
if (!updateToLocation(Loc))
return InsertPointTy();
@@ -2620,7 +2622,15 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::createTask(
Builder.CreateStore(Priority, CmplrData);
}
- Value *DepArray = emitTaskDependencies(*this, Dependencies);
+ Value *DepArray = nullptr;
+ Value *NumDeps = nullptr;
+ if (Dependencies.DepArray) {
+ DepArray = Dependencies.DepArray;
+ NumDeps = Dependencies.NumDeps;
+ } else if (!Dependencies.Deps.empty()) {
+ DepArray = emitTaskDependencies(*this, Dependencies.Deps);
+ NumDeps = Builder.getInt32(Dependencies.Deps.size());
+ }
// In the presence of the `if` clause, the following IR is generated:
// ...
@@ -2651,12 +2661,12 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::createTask(
&ElseTI);
Builder.SetInsertPoint(ElseTI);
- if (Dependencies.size()) {
+ if (DepArray) {
Function *TaskWaitFn =
getOrCreateRuntimeFunctionPtr(OMPRTL___kmpc_omp_wait_deps);
createRuntimeFunctionCall(
TaskWaitFn,
- {Ident, ThreadID, Builder.getInt32(Dependencies.size()), DepArray,
+ {Ident, ThreadID, NumDeps, DepArray,
ConstantInt::get(Builder.getInt32Ty(), 0),
ConstantPointerNull::get(PointerType::getUnqual(M.getContext()))});
}
@@ -2675,13 +2685,13 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::createTask(
Builder.SetInsertPoint(ThenTI);
}
- if (Dependencies.size()) {
+ if (DepArray) {
Function *TaskFn =
getOrCreateRuntimeFunctionPtr(OMPRTL___kmpc_omp_task_with_deps);
createRuntimeFunctionCall(
TaskFn,
- {Ident, ThreadID, TaskData, Builder.getInt32(Dependencies.size()),
- DepArray, ConstantInt::get(Builder.getInt32Ty(), 0),
+ {Ident, ThreadID, TaskData, NumDeps, DepArray,
+ ConstantInt::get(Builder.getInt32Ty(), 0),
ConstantPointerNull::get(PointerType::getUnqual(M.getContext()))});
} else {
@@ -8791,8 +8801,8 @@ static Error emitTargetOutlinedFunction(
OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::emitTargetTask(
TargetTaskBodyCallbackTy TaskBodyCB, Value *DeviceID, Value *RTLoc,
OpenMPIRBuilder::InsertPointTy AllocaIP,
- const SmallVector<llvm::OpenMPIRBuilder::DependData> &Dependencies,
- const TargetDataRTArgs &RTArgs, bool HasNoWait) {
+ const DependenciesInfo &Dependencies, const TargetDataRTArgs &RTArgs,
+ bool HasNoWait) {
// The following explains the code-gen scenario for the `target` directive. A
// similar scneario is followed for other device-related directives (e.g.
@@ -9109,7 +9119,15 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::emitTargetTask(
}
}
- Value *DepArray = emitTaskDependencies(*this, Dependencies);
+ Value *DepArray = nullptr;
+ Value *NumDeps = nullptr;
+ if (Dependencies.DepArray) {
+ DepArray = Dependencies.DepArray;
+ NumDeps = Dependencies.NumDeps;
+ } else if (!Dependencies.Deps.empty()) {
+ DepArray = emitTaskDependencies(*this, Dependencies.Deps);
+ NumDeps = Builder.getInt32(Dependencies.Deps.size());
+ }
// ---------------------------------------------------------------
// V5.2 13.8 target construct
@@ -9126,7 +9144,7 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::emitTargetTask(
createRuntimeFunctionCall(
TaskWaitFn,
{/*loc_ref=*/Ident, /*gtid=*/ThreadID,
- /*ndeps=*/Builder.getInt32(Dependencies.size()),
+ /*ndeps=*/NumDeps,
/*dep_list=*/DepArray,
/*ndeps_noalias=*/ConstantInt::get(Builder.getInt32Ty(), 0),
/*noalias_dep_list=*/
@@ -9149,8 +9167,8 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::emitTargetTask(
getOrCreateRuntimeFunctionPtr(OMPRTL___kmpc_omp_task_with_deps);
createRuntimeFunctionCall(
TaskFn,
- {Ident, ThreadID, TaskData, Builder.getInt32(Dependencies.size()),
- DepArray, ConstantInt::get(Builder.getInt32Ty(), 0),
+ {Ident, ThreadID, TaskData, NumDeps, DepArray,
+ ConstantInt::get(Builder.getInt32Ty(), 0),
ConstantPointerNull::get(PointerType::getUnqual(M.getContext()))});
} else {
// Emit the @__kmpc_omp_task runtime call to spawn the task
@@ -9185,19 +9203,19 @@ Error OpenMPIRBuilder::emitOffloadingArraysAndArgs(
return Error::success();
}
-static void emitTargetCall(
- OpenMPIRBuilder &OMPBuilder, IRBuilderBase &Builder,
- OpenMPIRBuilder::InsertPointTy AllocaIP,
- OpenMPIRBuilder::TargetDataInfo &Info,
- const OpenMPIRBuilder::TargetKernelDefaultAttrs &DefaultAttrs,
- const OpenMPIRBuilder::TargetKernelRuntimeAttrs &RuntimeAttrs,
- Value *IfCond, Function *OutlinedFn, Constant *OutlinedFnID,
- SmallVectorImpl<Value *> &Args,
- OpenMPIRBuilder::GenMapInfoCallbackTy GenMapInfoCB,
- OpenMPIRBuilder::CustomMapperCallbackTy CustomMapperCB,
- const SmallVector<llvm::OpenMPIRBuilder::DependData> &Dependencies,
- bool HasNoWait, Value *DynCGroupMem,
- OMPDynGroupprivateFallbackType DynCGroupMemFallback) {
+static void
+emitTargetCall(OpenMPIRBuilder &OMPBuilder, IRBuilderBase &Builder,
+ OpenMPIRBuilder::InsertPointTy AllocaIP,
+ OpenMPIRBuilder::TargetDataInfo &Info,
+ const OpenMPIRBuilder::TargetKernelDefaultAttrs &DefaultAttrs,
+ const OpenMPIRBuilder::TargetKernelRuntimeAttrs &RuntimeAttrs,
+ Value *IfCond, Function *OutlinedFn, Constant *OutlinedFnID,
+ SmallVectorImpl<Value *> &Args,
+ OpenMPIRBuilder::GenMapInfoCallbackTy GenMapInfoCB,
+ OpenMPIRBuilder::CustomMapperCallbackTy CustomMapperCB,
+ const OpenMPIRBuilder::DependenciesInfo &Dependencies,
+ bool HasNoWait, Value *DynCGroupMem,
+ OMPDynGroupprivateFallbackType DynCGroupMemFallback) {
// Generate a function call to the host fallback implementation of the target
// region. This is called by the host when no offload entry was generated for
// the target region and when the offloading call fails at runtime.
@@ -9212,7 +9230,7 @@ static void emitTargetCall(
return Builder.saveIP();
};
- bool HasDependencies = Dependencies.size() > 0;
+ bool HasDependencies = !Dependencies.empty();
bool RequiresOuterTargetTask = HasNoWait || HasDependencies;
OpenMPIRBuilder::TargetKernelArgs KArgs;
@@ -9390,9 +9408,9 @@ OpenMPIRBuilder::InsertPointOrErrorTy OpenMPIRBuilder::createTarget(
SmallVectorImpl<Value *> &Inputs, GenMapInfoCallbackTy GenMapInfoCB,
OpenMPIRBuilder::TargetBodyGenCallbackTy CBFunc,
OpenMPIRBuilder::TargetGenArgAccessorsCallbackTy ArgAccessorFuncCB,
- CustomMapperCallbackTy CustomMapperCB,
- const SmallVector<DependData> &Dependencies, bool HasNowait,
- Value *DynCGroupMem, OMPDynGroupprivateFallbackType DynCGroupMemFallback) {
+ CustomMapperCallbackTy CustomMapperCB, const DependenciesInfo &Dependencies,
+ bool HasNowait, Value *DynCGroupMem,
+ OMPDynGroupprivateFallbackType DynCGroupMemFallback) {
if (!updateToLocation(Loc))
return InsertPointTy();
diff --git a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
index 7d25ac9cadedb..4f17b54b07e83 100644
--- a/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
+++ b/llvm/unittests/Frontend/OpenMPIRBuilderTest.cpp
@@ -7350,7 +7350,8 @@ TEST_F(OpenMPIRBuilderTest, CreateTaskDepend) {
OMPBuilder.createTask(
Loc, InsertPointTy(AllocaBB, AllocaBB->getFirstInsertionPt()),
BodyGenCB,
- /*Tied=*/false, /*Final*/ nullptr, /*IfCondition*/ nullptr, DDS));
+ /*Tied=*/false, /*Final*/ nullptr, /*IfCondition*/ nullptr,
+ OpenMPIRBuilder::DependenciesInfo{std::move(DDS)}));
Builder.restoreIP(AfterIP);
OMPBuilder.finalize();
Builder.CreateRetVoid();
diff --git a/mlir/lib/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.cpp b/mlir/lib/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.cpp
index 9f8c13bbe9060..dc45e8cf3b963 100644
--- a/mlir/lib/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.cpp
+++ b/mlir/lib/Target/LLVMIR/Dialect/OpenMP/OpenMPToLLVMIRTranslation.cpp
@@ -357,11 +357,6 @@ static LogicalResult checkImplementationStatus(Operation &op) {
if (!op.getDependVars().empty() || op.getDependKinds())
result = todo("depend");
};
- auto checkDependIteratorModifier = [&todo](auto op, LogicalResult &result) {
- if (!op.getDependIterated().empty() ||
- (op.getDependIteratedKinds() && !op.getDependIteratedKinds()->empty()))
- result = todo("depend with iterator modifier");
- };
auto checkHint = [](auto op, LogicalResult &) {
if (op.getHint())
op.emitWarning("hint clause discarded");
@@ -434,7 +429,6 @@ static LogicalResult checkImplementationStatus(Operation &op) {
})
.Case([&](omp::TaskOp op) {
checkAllocate(op, result);
- checkDependIteratorModifier(op, result);
checkInReduction(op, result);
})
.Case([&](omp::TaskgroupOp op) {
@@ -469,7 +463,6 @@ static LogicalResult checkImplementationStatus(Operation &op) {
.Case([&](omp::TargetOp op) {
checkAllocate(op, result);
checkBare(op, result);
- checkDependIteratorModifier(op, result);
checkInReduction(op, result);
checkThreadLimit(op, result);
})
@@ -2121,33 +2114,35 @@ convertOmpTeams(omp::TeamsOp op, llvm::IRBuilderBase &builder,
return success();
}
-static void
-buildDependData(std::optional<ArrayAttr> dependKinds, OperandRange dependVars,
- LLVM::ModuleTranslation &moduleTranslation,
- SmallVectorImpl<llvm::OpenMPIRBuilder::DependData> &dds) {
+static llvm::omp::RTLDependenceKindTy
+convertDependKind(mlir::omp::ClauseTaskDepend kind) {
+ switch (kind) {
+ case mlir::omp::ClauseTaskDepend::taskdependin:
+ return llvm::omp::RTLDependenceKindTy::DepIn;
+ // The OpenMP runtime requires that the codegen for 'depend' clause for
+ // 'out' dependency kind must be the same as codegen for 'depend' clause
+ // with 'inout' dependency.
+ case mlir::omp::ClauseTaskDepend::taskdependout:
+ case mlir::omp::ClauseTaskDepend::taskdependinout:
+ return llvm::omp::RTLDependenceKindTy::DepInOut;
+ case mlir::omp::ClauseTaskDepend::taskdependmutexinoutset:
+ return llvm::omp::RTLDependenceKindTy::DepMutexInOutSet;
+ case mlir::omp::ClauseTaskDepend::taskdependinoutset:
+ return llvm::omp::RTLDependenceKindTy::DepInOutSet;
+ }
+ llvm_unreachable("unhandled depend kind");
+}
+
+static void buildDependDataLocator(
+ std::optional<ArrayAttr> dependKinds, OperandRange dependVars,
+ LLVM::ModuleTranslation &moduleTranslation,
+ SmallVectorImpl<llvm::OpenMPIRBuilder::DependData> &dds) {
if (dependVars.empty())
return;
for (auto dep : llvm::zip(dependVars, dependKinds->getValue())) {
- llvm::omp::RTLDependenceKindTy type;
- switch (
- cast<mlir::omp::ClauseTaskDependAttr>(std::get<1>(dep)).getValue()) {
- case mlir::omp::ClauseTaskDepend::taskdependin:
- type = llvm::omp::RTLDependenceKindTy::DepIn;
- break;
- // The OpenMP runtime requires that the codegen for 'depend' clause for
- // 'out' dependency kind must be the same as codegen for 'depend' clause
- // with 'inout' dependency.
- case mlir::omp::ClauseTaskDepend::taskdependout:
- case mlir::omp::ClauseTaskDepend::taskdependinout:
- type = llvm::omp::RTLDependenceKindTy::DepInOut;
- break;
- case mlir::omp::ClauseTaskDepend::taskdependmutexinoutset:
- type = llvm::omp::RTLDependenceKindTy::DepMutexInOutSet;
- break;
- case mlir::omp::ClauseTaskDepend::taskdependinoutset:
- type = llvm:...
[truncated]
``````````
</details>
https://github.com/llvm/llvm-project/pull/190026
More information about the Mlir-commits
mailing list