[Mlir-commits] [mlir] [MLIR][SCFToOpenMP] Add num-threads option	(PR #74854)
    Michael Klemm 
    llvmlistbot at llvm.org
       
    Mon Dec 11 04:29:12 PST 2023
    
    
  
================
@@ -390,6 +392,12 @@ struct ParallelOpLowering : public OpRewritePattern<scf::ParallelOp> {
 
     // Create the parallel wrapper.
     auto ompParallel = rewriter.create<omp::ParallelOp>(loc);
+    if (numThreads > 1) {
+      rewriter.setInsertionPoint(ompParallel);
+      mlir::Value numThreadsVar = rewriter.create<LLVM::ConstantOp>(
+          loc, rewriter.getI32IntegerAttr(numThreads));
----------------
mjklemm wrote:
The entrypoint `__kmpc_push_num_threads` takes a `int32_t` for the number of threads coming from the `num_threads` clause.  So, a 32bit int might be fine. 
https://github.com/llvm/llvm-project/pull/74854
    
    
More information about the Mlir-commits
mailing list