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
|
//------------------------------------------------------------------------------
// GB_select_value_iso: select when A is iso and the op is VALUE*
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// A is iso and the operator is VALUE*.
// The VALUE* operators depend only on the value of A(i,j). Since A is iso,
// either all entries in A will be copied to C and thus C can be created as a
// shallow copy of A, or no entries from A will be copied to C and thus C is an
// empty matrix. The select factory is not needed, except to check the iso
// value via GB_select_bitmap.
// This method takes O(1) time and space.
#define GB_FREE_ALL \
{ \
GB_phybix_free (C) ; \
}
#include "select/GB_select.h"
#include "scalar/GB_Scalar_wrap.h"
GrB_Info GB_select_value_iso
(
GrB_Matrix C,
GrB_IndexUnaryOp op,
GrB_Matrix A,
int64_t ithunk,
const GB_void *restrict athunk,
const GB_void *restrict ythunk,
GB_Werk Werk
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
ASSERT (A->iso) ;
ASSERT (op->opcode >= GB_VALUENE_idxunop_code
&& op->opcode <= GB_VALUELE_idxunop_code)
ASSERT (C != NULL && (C->header_size == 0 || GBNSTATIC)) ;
//--------------------------------------------------------------------------
// determine if C is empty or a copy of A
//--------------------------------------------------------------------------
GrB_Info info ;
GrB_Type xtype = op->xtype ;
ASSERT (xtype != NULL) ;
const size_t xsize = xtype->size ;
// construct a scalar S containing the iso scalar of ((xtype) A)
struct GB_Scalar_opaque S_header ;
GrB_Scalar S ;
// wrap the iso-value of A in the scalar S, typecasted to xtype
// xscalar = (op->xtype) A->x
GB_void xscalar [GB_VLA(xsize)] ;
GB_cast_scalar (xscalar, xtype->code, A->x, A->type->code,
A->type->size) ;
S = GB_Scalar_wrap (&S_header, xtype, xscalar) ;
S->iso = false ; // but ensure S is not iso
ASSERT_SCALAR_OK (S, "iso scalar wrap", GB0) ;
// apply the select operator to the iso scalar S
GB_OK (GB_select_bitmap (C, false, op, false, (GrB_Matrix) S, ithunk,
athunk, ythunk, Werk)) ;
ASSERT_MATRIX_OK (C, "C from iso scalar test", GB0) ;
bool C_empty = (GB_nnz (C) == 0) ;
GB_phybix_free (C) ;
//--------------------------------------------------------------------------
// construct C: either an empty matrix, or a copy of A
//--------------------------------------------------------------------------
if (C_empty)
{
// C is an empty: create a new empty matrix (not a shallow copy of A)
// determine the p_is_32, j_is_32, and i_is_32 settings for C
bool Cp_is_32, Cj_is_32, Ci_is_32 ;
GB_determine_pji_is_32 (&Cp_is_32, &Cj_is_32, &Ci_is_32,
GxB_AUTO_SPARSITY, 0, A->vlen, A->vdim, Werk) ;
return (GB_new (&C, // existing header
A->type, A->vlen, A->vdim, GB_ph_calloc, true,
GxB_AUTO_SPARSITY, GB_Global_hyper_switch_get ( ), 1,
Cp_is_32, Cj_is_32, Ci_is_32)) ;
}
else
{
// C is a shallow copy of A with all the same entries as A
return (GB_shallow_copy (C, true, A, Werk)) ;
}
}
|
