diff --git a/cpu/kernelcreation.py b/cpu/kernelcreation.py index cbea79d7daa20bb7012e3a4b0851a90d88510afe..3c2621404ecb6cffba62d7ec7410d22af0ae7af2 100644 --- a/cpu/kernelcreation.py +++ b/cpu/kernelcreation.py @@ -72,11 +72,11 @@ def addOpenMP(astNode, schedule="static", numThreads=None): """ assert type(astNode) is ast.KernelFunction body = astNode.body - wrapperBlock = ast.PragmaBlock('#pragma omp parallel', body.takeChildNodes()) + threadsClause = "" if numThreads is None else " num_threads(%s)" % (numThreads,) + wrapperBlock = ast.PragmaBlock('#pragma omp parallel' + threadsClause, body.takeChildNodes()) body.append(wrapperBlock) outerLoops = [l for l in body.atoms(ast.LoopOverCoordinate) if l.isOutermostLoop] assert outerLoops, "No outer loop found" assert len(outerLoops) <= 1, "More than one outer loop found. Which one should be parallelized?" - threadsClause = "" if numThreads is None else " num_threads(%s)" % (numThreads,) - outerLoops[0].prefixLines.append("#pragma omp for schedule(%s)%s" % (schedule,threadsClause)) + outerLoops[0].prefixLines.append("#pragma omp for schedule(%s)" % (schedule,))