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
|
//------------------------------------------------------------------------------
// GB_bitmap_selector: select entries from a bitmap or full matrix
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
#include "GB_select.h"
#include "GB_sel__include.h"
#define GB_FREE_ALL ;
GrB_Info GB_bitmap_selector
(
GrB_Matrix C, // output matrix, static header
const bool C_iso, // if true, C is iso
GB_Opcode opcode, // selector/idxunop opcode
const GB_Operator op,
const bool flipij, // if true, flip i and j for user operator
GrB_Matrix A, // input matrix
const int64_t ithunk, // (int64_t) Thunk, if Thunk is NULL
const GB_void *restrict athunk, // (A->type) Thunk
const GB_void *restrict ythunk, // (op->utype) Thunk
GB_Context Context
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
GrB_Info info ;
ASSERT_MATRIX_OK (A, "A for bitmap selector", GB0) ;
ASSERT (GB_IS_BITMAP (A) || GB_as_if_full (A)) ;
ASSERT (opcode != GB_NONZOMBIE_selop_code) ;
ASSERT (C != NULL && (C->static_header || GBNSTATIC)) ;
//--------------------------------------------------------------------------
// get A
//--------------------------------------------------------------------------
int64_t anz = GB_nnz_held (A) ;
const size_t asize = A->type->size ;
const GB_Type_code acode = A->type->code ;
//--------------------------------------------------------------------------
// allocate C
//--------------------------------------------------------------------------
// C->b and C->x are malloc'd, not calloc'd
// set C->iso = C_iso OK
GB_OK (GB_new_bix (&C, // always bitmap, existing header
A->type, A->vlen, A->vdim, GB_Ap_calloc, true,
GxB_BITMAP, false, A->hyper_switch, -1, anz, true, C_iso, Context)) ;
int64_t cnvals ;
//--------------------------------------------------------------------------
// determine the number of threads to use
//--------------------------------------------------------------------------
GB_GET_NTHREADS_MAX (nthreads_max, chunk, Context) ;
int nthreads = GB_nthreads (anz, chunk, nthreads_max) ;
//--------------------------------------------------------------------------
// set the iso value of C
//--------------------------------------------------------------------------
if (C_iso)
{
GB_iso_select (C->x, opcode, athunk, A->x, acode, asize) ;
}
//--------------------------------------------------------------------------
// launch the switch factory to select the entries
//--------------------------------------------------------------------------
#define GB_BITMAP_SELECTOR
#define GB_selbit(opname,aname) GB (_sel_bitmap_ ## opname ## aname)
#define GB_SEL_WORKER(opname,aname,atype) \
{ \
GB_selbit (opname, aname) (C->b, (atype *) C->x, &cnvals, A, \
flipij, ithunk, (atype *) athunk, ythunk, op, nthreads) ; \
} \
break ;
const GB_Type_code typecode = (A->iso) ? GB_ignore_code : acode ;
#include "GB_select_factory.c"
//--------------------------------------------------------------------------
// return result
//--------------------------------------------------------------------------
C->nvals = cnvals ;
C->magic = GB_MAGIC ;
ASSERT_MATRIX_OK (C, "C from bitmap selector", GB0) ;
return (GrB_SUCCESS) ;
}
|
