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
|
//------------------------------------------------------------------------------
// GB_AxB_semiring_builtin: determine if semiring is built-in
//------------------------------------------------------------------------------
// SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
// SPDX-License-Identifier: Apache-2.0
//------------------------------------------------------------------------------
// Determine if A*B uses a built-in semiring, and if so, determine the
// opcodes and type codes of the semiring.
// This function is not used by the CUDA jitified kernels, since they can
// typecast the entries in the matrices A and B to the types of x and y of the
// operator, as needed.
#include "GB_mxm.h"
#include "GB_binop.h"
bool GB_AxB_semiring_builtin // true if semiring is builtin
(
// inputs:
const GrB_Matrix A,
const bool A_is_pattern, // true if only the pattern of A is used
const GrB_Matrix B,
const bool B_is_pattern, // true if only the pattern of B is used
const GrB_Semiring semiring, // semiring that defines C=A*B
const bool flipxy, // true if z=fmult(y,x), flipping x and y
// outputs:
GB_Opcode *mult_binop_code, // multiply opcode
GB_Opcode *add_binop_code, // add opcode
GB_Type_code *xcode, // type code for x input
GB_Type_code *ycode, // type code for y input
GB_Type_code *zcode // type code for z output
)
{
//--------------------------------------------------------------------------
// check inputs
//--------------------------------------------------------------------------
// A and B may be aliased
GrB_BinaryOp add = semiring->add->op ; // add operator
GrB_BinaryOp mult = semiring->multiply ; // multiply operator
// add is a monoid
ASSERT (add->xtype == add->ztype && add->ytype == add->ztype) ;
ASSERT (!GB_OP_IS_POSITIONAL (add)) ;
// in a semiring, the ztypes of add and mult are always the same:
ASSERT (add->ztype == mult->ztype) ;
// The conditions above are true for any semiring and any A and B, whether
// or not this function handles the semiring as hard-coded. Now return for
// cases this function does not handle.
(*mult_binop_code) = GB_NOP_code ;
(*xcode) = GB_ignore_code ;
(*ycode) = GB_ignore_code ;
(*zcode) = GB_ignore_code ;
//--------------------------------------------------------------------------
// check the monoid
//--------------------------------------------------------------------------
(*add_binop_code) = add->opcode ;
ASSERT (GB_IS_BINARYOP_CODE (*add_binop_code)) ;
if (*add_binop_code == GB_USER_binop_code)
{
// semiring has a user-defined add operator for its monoid
return (false) ;
}
//--------------------------------------------------------------------------
// rename redundant boolean monoids
//--------------------------------------------------------------------------
if (add->ztype->code == GB_BOOL_code)
{
// Only the LAND, LOR, LXOR, and EQ monoids remain if z is
// boolean. MIN, MAX, PLUS, and TIMES are renamed.
(*add_binop_code) = GB_boolean_rename (*add_binop_code) ;
}
//--------------------------------------------------------------------------
// check the multiply operator
//--------------------------------------------------------------------------
if (!GB_binop_builtin (A->type, A_is_pattern, B->type, B_is_pattern,
mult, flipxy, mult_binop_code, xcode, ycode, zcode))
{
return (false) ;
}
//--------------------------------------------------------------------------
// rename to ANY_PAIR
//--------------------------------------------------------------------------
if ((*mult_binop_code) == GB_PAIR_binop_code)
{
if (((*add_binop_code) == GB_EQ_binop_code) ||
((*add_binop_code) == GB_LAND_binop_code) ||
((*add_binop_code) == GB_BAND_binop_code) ||
((*add_binop_code) == GB_LOR_binop_code) ||
((*add_binop_code) == GB_BOR_binop_code) ||
((*add_binop_code) == GB_MAX_binop_code) ||
((*add_binop_code) == GB_MIN_binop_code) ||
((*add_binop_code) == GB_TIMES_binop_code))
// rename to ANY_PAIR
(*add_binop_code) = GB_ANY_binop_code ;
}
return (true) ;
}
|
