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 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83
|
//------------------------------------------------------------------------------
// GB_red: hard-coded functions for reductions
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#include "GB_control.h"
#if defined (GxB_NO_INT16)
#define GB_TYPE_ENABLED 0
#else
#define GB_TYPE_ENABLED 1
#endif
#if GB_TYPE_ENABLED
#include "GB.h"
#include "FactoryKernels/GB_red__include.h"
// reduction operator and type:
#define GB_UPDATE(z,a) z += a
#define GB_ADD(z,zin,a) z = zin + a
#define GB_GETA_AND_UPDATE(z,Ax,p) z += Ax [p]
// A matrix (no typecasting to Z type here)
#define GB_A_TYPE int16_t
#define GB_DECLAREA(aij) int16_t aij
#define GB_GETA(aij,Ax,pA,A_iso) aij = Ax [pA]
// monoid properties:
#define GB_Z_TYPE int16_t
#define GB_DECLARE_IDENTITY(z) int16_t z = 0
#define GB_DECLARE_IDENTITY_CONST(z) const int16_t z = 0
// panel size
#define GB_PANEL 64
// disable this operator and use the generic case if these conditions hold
#if (defined(GxB_NO_PLUS) || defined(GxB_NO_INT16) || defined(GxB_NO_PLUS_INT16))
#define GB_DISABLE 1
#else
#define GB_DISABLE 0
#endif
#include "monoid/include/GB_monoid_shared_definitions.h"
//------------------------------------------------------------------------------
// reduce to a non-iso matrix to scalar, for monoids only
//------------------------------------------------------------------------------
GrB_Info GB (_red__plus_int16)
(
GB_Z_TYPE *result,
const GrB_Matrix A,
GB_void *restrict W_space,
bool *restrict F,
int ntasks,
int nthreads
)
{
#if GB_DISABLE
return (GrB_NO_VALUE) ;
#else
GB_Z_TYPE z = (*result) ;
GB_Z_TYPE *restrict W = (GB_Z_TYPE *) W_space ;
if (A->nzombies > 0 || GB_IS_BITMAP (A))
{
#include "reduce/template/GB_reduce_to_scalar_template.c"
}
else
{
#include "reduce/template/GB_reduce_panel.c"
}
(*result) = z ;
return (GrB_SUCCESS) ;
#endif
}
#else
GB_EMPTY_PLACEHOLDER
#endif
|