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