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_UINT16)
#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 uint16_t
#define GB_DECLAREA(aij) uint16_t aij
#define GB_GETA(aij,Ax,pA,A_iso) aij = Ax [pA]
// monoid properties:
#define GB_Z_TYPE uint16_t
#define GB_DECLARE_IDENTITY(z) uint16_t z = 0
#define GB_DECLARE_IDENTITY_CONST(z) const uint16_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_UINT16) || defined(GxB_NO_PLUS_UINT16))
#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_uint16)
(
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
|
