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
|
//------------------------------------------------------------------------------
// GB_select_generic_phase1.c: count entries for C=select(A,thunk)
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// A and C are sparse or hypersparse
#include "select/GB_select.h"
#include "include/GB_unused.h"
GrB_Info GB_select_generic_phase1
(
GrB_Matrix C,
uint64_t *restrict Wfirst,
uint64_t *restrict Wlast,
const GrB_Matrix A,
const bool flipij,
const GB_void *restrict ythunk,
const GrB_IndexUnaryOp op,
const int64_t *A_ek_slicing,
const int A_ntasks,
const int A_nthreads
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
GB_Opcode opcode = op->opcode ;
ASSERT (GB_IS_SPARSE (A) || GB_IS_HYPERSPARSE (A)) ;
ASSERT (GB_IS_INDEXUNARYOP_CODE (opcode)) ;
ASSERT (!GB_IS_INDEXUNARYOP_CODE_POSITIONAL (opcode)) ;
ASSERT (opcode != GB_NONZOMBIE_idxunop_code) ;
//--------------------------------------------------------------------------
// phase1: generic entry selector
//--------------------------------------------------------------------------
ASSERT_TYPE_OK (op->xtype, "op->xtype", GB0) ;
GB_Type_code zcode = op->ztype->code ;
GB_Type_code xcode = op->xtype->code ;
GB_Type_code acode = A->type->code ;
size_t zsize = op->ztype->size ;
size_t xsize = op->xtype->size ;
size_t asize = A->type->size ;
GxB_index_unary_function fkeep = op->idxunop_function ;
GB_cast_function cast_Z_to_bool, cast_A_to_X ;
#define GB_GENERIC
#define GB_A_TYPE GB_void
#include "select/include/GB_select_shared_definitions.h"
if (A->iso)
{
//----------------------------------------------------------------------
// A is iso
//----------------------------------------------------------------------
// x = (xtype) Ax [0]
GB_void x [GB_VLA(xsize)] ;
GB_cast_scalar (x, xcode, A->x, acode, asize) ;
if (op->ztype == GrB_BOOL)
{
//------------------------------------------------------------------
// A is iso and z is bool
//------------------------------------------------------------------
#define GB_TEST_VALUE_OF_ENTRY(keep,p) \
bool keep ; \
fkeep (&keep, x, flipij ? j : i, flipij ? i : j, ythunk) ;
#include "select/template/GB_select_entry_phase1_template.c"
}
else
{
//------------------------------------------------------------------
// A is iso and z requires typecasting
//------------------------------------------------------------------
cast_Z_to_bool = GB_cast_factory (GB_BOOL_code, zcode) ;
#undef GB_TEST_VALUE_OF_ENTRY
#define GB_TEST_VALUE_OF_ENTRY(keep,p) \
bool keep ; \
GB_void z [GB_VLA(zsize)] ; \
fkeep (z, x, flipij ? j : i, flipij ? i : j, ythunk) ; \
cast_Z_to_bool (&keep, z, zsize) ;
#include "select/template/GB_select_entry_phase1_template.c"
}
}
else
{
if (op->ztype == GrB_BOOL && op->xtype == A->type)
{
//------------------------------------------------------------------
// A is non-iso and no typecasting is required
//------------------------------------------------------------------
#undef GB_TEST_VALUE_OF_ENTRY
#define GB_TEST_VALUE_OF_ENTRY(keep,p) \
bool keep ; \
fkeep (&keep, Ax +(p)*asize, \
flipij ? j : i, flipij ? i : j, ythunk) ;
#include "select/template/GB_select_entry_phase1_template.c"
}
else
{
//------------------------------------------------------------------
// A is non-iso and typecasting is required
//------------------------------------------------------------------
cast_A_to_X = GB_cast_factory (xcode, acode) ;
cast_Z_to_bool = GB_cast_factory (GB_BOOL_code, zcode) ;
#undef GB_TEST_VALUE_OF_ENTRY
#define GB_TEST_VALUE_OF_ENTRY(keep,p) \
bool keep ; \
GB_void z [GB_VLA(zsize)] ; \
GB_void x [GB_VLA(xsize)] ; \
cast_A_to_X (x, Ax +(p)*asize, asize) ; \
fkeep (z, x, flipij ? j : i, flipij ? i : j, ythunk) ; \
cast_Z_to_bool (&keep, z, zsize) ;
#include "select/template/GB_select_entry_phase1_template.c"
}
}
return (GrB_SUCCESS) ;
}
|
