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 52 53 54 55 56 57
|
//------------------------------------------------------------------------------
// gbchunk: get/set the chunk size to use in GraphBLAS
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// Usage:
// chunk = gbchunk ;
// chunk = gbchunk (chunk) ;
#include "gb_interface.h"
#define USAGE "usage: c = GrB.chunk ; or GrB.chunk (c)"
void mexFunction
(
int nargout,
mxArray *pargout [ ],
int nargin,
const mxArray *pargin [ ]
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
gb_usage (nargin <= 1 && nargout <= 1, USAGE) ;
//--------------------------------------------------------------------------
// set the chunk, if requested
//--------------------------------------------------------------------------
double c ;
if (nargin > 0)
{
// set the chunk
CHECK_ERROR (!gb_mxarray_is_scalar (pargin [0]),
"input must be a scalar") ;
c = (double) mxGetScalar (pargin [0]) ;
OK (GxB_Global_Option_set (GxB_CHUNK, c)) ;
}
//--------------------------------------------------------------------------
// return the chunk
//--------------------------------------------------------------------------
OK (GxB_Global_Option_get (GxB_CHUNK, &c)) ;
pargout [0] = mxCreateDoubleScalar (c) ;
GB_WRAPUP ;
}
|