Line data    Source code

       1             : //------------------------------------------------------------------------------
       2             : // LAGraph_SetNumThreads: set the # of threads to use
       3             : //------------------------------------------------------------------------------
       4             : 
       5             : // LAGraph, (c) 2019-2022 by The LAGraph Contributors, All Rights Reserved.
       6             : // SPDX-License-Identifier: BSD-2-Clause
       7             : //
       8             : // For additional details (including references to third party source code and
       9             : // other files) see the LICENSE file or contact permission@sei.cmu.edu. See
      10             : // Contributors.txt for a full list of contributors. Created, in part, with
      11             : // funding and support from the U.S. Government (see Acknowledgments.txt file).
      12             : // DM22-0790
      13             : 
      14             : // Contributed by Timothy A. Davis, Texas A&M University
      15             : 
      16             : //------------------------------------------------------------------------------
      17             : 
      18             : #include "LG_internal.h"
      19             : 
      20          12 : int LAGraph_SetNumThreads
      21             : (
      22             :     // input:
      23             :     int nthreads_outer,
      24             :     int nthreads_inner,
      25             :     char *msg
      26             : )
      27             : {
      28             : 
      29             :     //--------------------------------------------------------------------------
      30             :     // check inputs
      31             :     //--------------------------------------------------------------------------
      32             : 
      33          12 :     LG_CLEAR_MSG ;
      34             : 
      35             :     //--------------------------------------------------------------------------
      36             :     // set number of threads to use
      37             :     //--------------------------------------------------------------------------
      38             : 
      39          12 :     nthreads_outer = LAGRAPH_MAX (nthreads_outer, 1) ;
      40          12 :     nthreads_inner = LAGRAPH_MAX (nthreads_inner, 1) ;
      41             : 
      42             :     // set # of GraphBLAS threads
      43          12 :     GRB_TRY (LG_SET_NTHREADS (nthreads_inner)) ;
      44             : 
      45             :     // set # of LAGraph threads
      46          12 :     LG_nthreads_outer = nthreads_outer ;    // for LAGraph itself, if nested
      47             :                                             // regions call GraphBLAS
      48          12 :     LG_nthreads_inner = nthreads_inner ;    // for lower-level parallelism
      49             : 
      50          12 :     return (GrB_SUCCESS) ;
      51             : }