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
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
|
//------------------------------------------------------------------------------
// GB_AxB_dot4_cij.c: C(i,j) += A(:,i)'*B(:,j) for dot4 method
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// A is sparse or hypersparse, B is full or bitmap, and C is full
{
//--------------------------------------------------------------------------
// get C(i,j)
//--------------------------------------------------------------------------
// future:: allow for the use of any accum in the JIT: set cij = identity,
// and then use the accum when done.
const int64_t pC = i + pC_start ; // C(i,j) is at Cx [pC]
GB_C_TYPE GB_GET4C (cij, pC) ; // cij = Cx [pC]
//--------------------------------------------------------------------------
// C(i,j) += A (:,i)*B(:,j): a single dot product
//--------------------------------------------------------------------------
#if ( GB_B_IS_FULL )
{
//----------------------------------------------------------------------
// A is sparse/hyper and B is full
//----------------------------------------------------------------------
#if GB_IS_LXOR_PAIR_SEMIRING
{
// (boolean XOR monoid)_PAIR semiring
uint64_t t = ((uint64_t) cij) + ainz ;
cij = (GB_C_TYPE) (t & 0x1L) ;
}
#elif GB_IS_PLUS_PAIR_8_SEMIRING
{
// (PLUS int8, uint8 monoids)_PAIR semirings
// only used by the JIT if int8 and uint8 FactoryKernels disabled
uint64_t t = ((uint64_t) cij) + ainz ;
cij = (GB_C_TYPE) (t & 0xFFL) ;
}
#elif GB_IS_PLUS_PAIR_16_SEMIRING
{
// (PLUS int16, uint16 monoids)_PAIR semirings
// only used by the JIT if int16 and uint16 FactoryKernels disabled
uint64_t t = ((uint64_t) cij) + ainz ;
cij = (GB_C_TYPE) (t & 0xFFFFL) ;
}
#elif GB_IS_PLUS_PAIR_32_SEMIRING
{
// (PLUS int32, uint32 monoids)_PAIR semirings
uint64_t t = ((uint64_t) cij) + ainz ;
cij = (GB_C_TYPE) (t & 0xFFFFFFFFL) ;
}
#elif GB_IS_PLUS_PAIR_BIG_SEMIRING
{
// (PLUS int64, uint64, float, or double)_PAIR semirings
cij += (GB_C_TYPE) ainz ;
}
#elif GB_IS_MIN_FIRSTJ_SEMIRING
{
// MIN_FIRSTJ semiring: take the 1st entry in A(:,i)
if (ainz > 0)
{
int64_t k = GB_IGET (Ai, pA) + GB_OFFSET ;
cij = GB_IMIN (cij, k) ;
}
}
#elif GB_IS_MAX_FIRSTJ_SEMIRING
{
// MAX_FIRSTJ semiring: take last entry in A(:,i)
if (ainz > 0)
{
int64_t k = GB_IGET (Ai, pA_end-1) + GB_OFFSET ;
cij = GB_IMAX (cij, k) ;
}
}
#else
{
GB_PRAGMA_SIMD_REDUCTION_MONOID (cij)
for (int64_t p = pA ; p < pA_end ; p++)
{
int64_t k = GB_IGET (Ai, p) ;
GB_DOT (k, p, pB+k) ; // cij += A(k,i)*B(k,j)
}
}
#endif
}
#else
{
//----------------------------------------------------------------------
// A is sparse/hyper and B is bitmap
//----------------------------------------------------------------------
#if GB_IS_MIN_FIRSTJ_SEMIRING
{
// MIN_FIRSTJ semiring: take the first entry
for (int64_t p = pA ; p < pA_end ; p++)
{
int64_t k = GB_IGET (Ai, p) ;
if (Bb [pB+k])
{
cij = GB_IMIN (cij, k + GB_OFFSET) ;
break ;
}
}
}
#elif GB_IS_MAX_FIRSTJ_SEMIRING
{
// MAX_FIRSTJ semiring: take the last entry
for (int64_t p = pA_end-1 ; p >= pA ; p--)
{
int64_t k = GB_IGET (Ai, p) ;
if (Bb [pB+k])
{
cij = GB_IMAX (cij, k + GB_OFFSET) ;
break ;
}
}
}
#else
{
GB_PRAGMA_SIMD_REDUCTION_MONOID (cij)
for (int64_t p = pA ; p < pA_end ; p++)
{
int64_t k = GB_IGET (Ai, p) ;
if (Bb [pB+k])
{
GB_DOT (k, p, pB+k) ; // cij+=A(k,i)*B(k,j)
}
}
}
#endif
}
#endif
//--------------------------------------------------------------------------
// save C(i,j)
//--------------------------------------------------------------------------
// future:: add the accum here for the JIT kernel (arbitrary accum
// and typecasting)
Cx [pC] = cij ;
}
|
