File: codegen_axb_template.m

package info (click to toggle)
suitesparse 1%3A7.10.1%2Bdfsg-1
  • links: PTS, VCS
  • area: main
  • in suites: forky, sid, trixie
  • size: 254,920 kB
  • sloc: ansic: 1,134,743; cpp: 46,133; makefile: 4,875; fortran: 2,087; java: 1,826; sh: 996; ruby: 725; python: 495; asm: 371; sed: 166; awk: 44
file content (145 lines) | stat: -rw-r--r-- 9,301 bytes parent folder | download | duplicates (2)
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
function codegen_axb_template (multop, bmult, imult, fmult, dmult, fcmult, dcmult, no_min_max_any_times_monoids)
%CODEGEN_AXB_TEMPLATE create a function for a semiring with a TxT->T multiplier

% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0

fprintf ('\n%-7s', multop) ;
if (nargin < 8)
    no_min_max_any_times_monoids = false ;
end

is_pair = isequal (multop, 'pair') ;
% the any_pair_iso semiring
if (is_pair)
    codegen_axb_method ('any', 'pair') ;
end

plusinf32 = 'INFINITY' ;
neginf32  = '(-INFINITY)' ;
plusinf64 = '((double) INFINITY)' ;
neginf64  = '((double) -INFINITY)' ;

% MIN monoid: integer types are terminal, float and double are not.
if (~no_min_max_any_times_monoids)
    update = 'warg = GB_IMIN (warg, targ)' ;
    addfunc = 'zarg = GB_IMIN (xarg, yarg)' ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'int8_t'  , 'int8_t'  , 'INT8_MAX'  , 'INT8_MIN'  ) ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'int16_t' , 'int16_t' , 'INT16_MAX' , 'INT16_MIN' ) ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'int32_t' , 'int32_t' , 'INT32_MAX' , 'INT32_MIN' ) ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'int64_t' , 'int64_t' , 'INT64_MAX' , 'INT64_MIN' ) ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'uint8_t' , 'uint8_t' , 'UINT8_MAX' , '0'         ) ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'uint16_t', 'uint16_t', 'UINT16_MAX', '0'         ) ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'uint32_t', 'uint32_t', 'UINT32_MAX', '0'         ) ;
    codegen_axb_method ('min', multop, update, addfunc, imult, 'uint64_t', 'uint64_t', 'UINT64_MAX', '0'         ) ;
    update = 'warg = fminf (warg, targ)' ;
    addfunc = 'zarg = fminf (xarg, yarg)' ;
    codegen_axb_method ('min', multop, update, addfunc, fmult, 'float'   , 'float'   , plusinf32   , [ ]         ) ;
    update = 'warg = fmin (warg, targ)' ;
    addfunc = 'zarg = fmin (xarg, yarg)' ;
    codegen_axb_method ('min', multop, update, addfunc, dmult, 'double'  , 'double'  , plusinf64   , [ ]         ) ;
end

% MAX monoid: integer types are terminal, float and double are not.
if (~no_min_max_any_times_monoids)
    update = 'warg = GB_IMAX (warg, targ)' ;
    addfunc = 'zarg = GB_IMAX (xarg, yarg)' ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'int8_t'  , 'int8_t'  , 'INT8_MIN'  , 'INT8_MAX'  ) ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'int16_t' , 'int16_t' , 'INT16_MIN' , 'INT16_MAX' ) ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'int32_t' , 'int32_t' , 'INT32_MIN' , 'INT32_MAX' ) ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'int64_t' , 'int64_t' , 'INT64_MIN' , 'INT64_MAX' ) ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'uint8_t' , 'uint8_t' , '0'         , 'UINT8_MAX' ) ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'uint16_t', 'uint16_t', '0'         , 'UINT16_MAX') ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'uint32_t', 'uint32_t', '0'         , 'UINT32_MAX') ;
    codegen_axb_method ('max', multop, update, addfunc, imult, 'uint64_t', 'uint64_t', '0'         , 'UINT64_MAX') ;
    % floating-point MAX must use unsigned integer puns for compare-and-swap
    update = 'warg = fmaxf (warg, targ)' ;
    addfunc = 'zarg = fmaxf (xarg, yarg)' ;
    codegen_axb_method ('max', multop, update, addfunc, fmult, 'float'   , 'float'   , neginf32    , [ ]         ) ;
    update = 'warg = fmax (warg, targ)' ;
    addfunc = 'zarg = fmax (xarg, yarg)' ;
    codegen_axb_method ('max', multop, update, addfunc, dmult, 'double'  , 'double'  , neginf64    , [ ]         ) ;
end

% ANY monoid: all are terminal.
if (~no_min_max_any_times_monoids && ~is_pair)
    update = 'warg = targ' ;
    addfunc = 'zarg = yarg' ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'int8_t'  , 'int8_t'  , '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'int16_t' , 'int16_t' , '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'int32_t' , 'int32_t' , '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'int64_t' , 'int64_t' , '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'uint8_t' , 'uint8_t' , '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'uint16_t', 'uint16_t', '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'uint32_t', 'uint32_t', '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, imult, 'uint64_t', 'uint64_t', '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, fmult, 'float'   , 'float'   , '0' , '(any value)') ;
    codegen_axb_method ('any', multop, update, addfunc, dmult, 'double'  , 'double'  , '0' , '(any value)') ;
    % complex case:
    id = 'GxB_CMPLXF(0,0)' ;
    codegen_axb_method ('any', multop, update, addfunc, fcmult, 'GxB_FC32_t', 'GxB_FC32_t', id, '(any value)') ;
    id = 'GxB_CMPLX(0,0)' ;
    codegen_axb_method ('any', multop, update, addfunc, dcmult, 'GxB_FC64_t', 'GxB_FC64_t', id, '(any value)') ;
end

% PLUS monoid: none are terminal.
update = 'warg += targ' ;
addfunc = 'zarg = xarg + yarg' ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'int8_t'  , 'int8_t'  , '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'uint8_t' , 'uint8_t' , '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'int16_t' , 'int16_t' , '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'uint16_t', 'uint16_t', '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'int32_t' , 'int32_t' , '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'uint32_t', 'uint32_t', '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'int64_t' , 'int64_t' , '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, imult, 'uint64_t', 'uint64_t', '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, fmult, 'float'   , 'float'   , '0', [ ]) ;
codegen_axb_method ('plus', multop, update, addfunc, dmult, 'double'  , 'double'  , '0', [ ]) ;
update = 'warg = GB_FC32_add (warg, targ)' ;
addfunc = 'zarg = GB_FC32_add (xarg, yarg)' ;
id = 'GxB_CMPLXF(0,0)' ;
codegen_axb_method ('plus', multop, update, addfunc, fcmult, 'GxB_FC32_t', 'GxB_FC32_t', id, [ ]) ;
update = 'warg = GB_FC64_add (warg, targ)' ;
addfunc = 'zarg = GB_FC64_add (xarg, yarg)' ;
id = 'GxB_CMPLX(0,0)' ;
codegen_axb_method ('plus', multop, update, addfunc, dcmult, 'GxB_FC64_t', 'GxB_FC64_t', id, [ ]) ;

% TIMES monoid: integers are terminal, float and double are not.
if (~no_min_max_any_times_monoids)
    update = 'warg *= targ' ;
    addfunc = 'zarg = xarg * yarg' ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'int8_t'  , 'int8_t'  , '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'uint8_t' , 'uint8_t' , '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'int16_t' , 'int16_t' , '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'uint16_t', 'uint16_t', '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'int32_t' , 'int32_t' , '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'uint32_t', 'uint32_t', '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'int64_t' , 'int64_t' , '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, imult, 'uint64_t', 'uint64_t', '1', '0') ;
    codegen_axb_method ('times', multop, update, addfunc, fmult, 'float'   , 'float'   , '1', [ ]) ;
    codegen_axb_method ('times', multop, update, addfunc, dmult, 'double'  , 'double'  , '1', [ ]) ;
    update = 'warg = GB_FC32_mul (warg, targ)' ;
    addfunc = 'zarg = GB_FC32_mul (xarg, yarg)' ;
    id = 'GxB_CMPLXF(1,0)' ;
    codegen_axb_method ('times', multop, update, addfunc, fcmult, 'GxB_FC32_t', 'GxB_FC32_t', id, [ ]) ;
    update = 'warg = GB_FC64_mul (warg, targ)' ;
    addfunc = 'zarg = GB_FC64_mul (xarg, yarg)' ;
    id = 'GxB_CMPLX(1,0)' ;
    codegen_axb_method ('times', multop, update, addfunc, dcmult, 'GxB_FC64_t', 'GxB_FC64_t', id, [ ]) ;
end

% boolean monoids: LOR, LAND are terminal; LXOR, EQ are not.
addfunc = 'zarg = xarg || yarg' ;
codegen_axb_method ('lor',  multop, 'warg |= targ', addfunc, bmult, 'bool', 'bool', 'false', 'true' ) ;
addfunc = 'zarg = xarg && yarg' ;
codegen_axb_method ('land', multop, 'warg &= targ', addfunc, bmult, 'bool', 'bool', 'true' , 'false') ;
addfunc = 'zarg = xarg != yarg' ;
codegen_axb_method ('lxor', multop, 'warg ^= targ', addfunc, bmult, 'bool', 'bool', 'false', [ ]    ) ;
if (~is_pair)
    addfunc = 'zarg = yarg' ;
    codegen_axb_method ('any', multop, 'warg = targ' , addfunc , bmult, 'bool', 'bool', '0'    , '(any value)') ;
end
update = 'warg = (warg == targ)' ;
addfunc = 'zarg = xarg == yarg' ;
codegen_axb_method ('eq',   multop, update,      addfunc, bmult, 'bool', 'bool', 'true' , [ ]    ) ;