[llvm-branch-commits] [clang] [llvm] [Clang][OMPX] Add the code generation for multi-dim `thread_limit` clause (PR #102717)
Alexey Bataev via llvm-branch-commits
llvm-branch-commits at lists.llvm.org
Sat Aug 10 06:25:34 PDT 2024
================
@@ -505,11 +505,14 @@ void OpenMPIRBuilder::getKernelArgsVector(TargetKernelArgs &KernelArgs,
Value *NumTeams3D =
Builder.CreateInsertValue(ZeroArray, KernelArgs.NumTeams[0], {0});
+ Value *NumThreads3D =
+ Builder.CreateInsertValue(ZeroArray, KernelArgs.NumThreads[0], {0});
for (unsigned I = 1; I < std::min(KernelArgs.NumTeams.size(), MaxDim); ++I)
NumTeams3D =
Builder.CreateInsertValue(NumTeams3D, KernelArgs.NumTeams[I], {I});
- Value *NumThreads3D =
- Builder.CreateInsertValue(ZeroArray, KernelArgs.NumThreads, {0});
+ for (unsigned I = 1; I < std::min(KernelArgs.NumThreads.size(), MaxDim); ++I)
----------------
alexey-bataev wrote:
`for (unsigned I : seq<unsigned>(1, std::min(KernelArgs.NumThreads.size(), MaxDim)))`
https://github.com/llvm/llvm-project/pull/102717
More information about the llvm-branch-commits
mailing list