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
|
function codegen_uop_method (uop, op, ztype, xtype)
%CODEGEN_UOP_METHOD create a function to compute C=uop(A)
%
% codegen_uop_method (uop, op, ztype, xtype)
%
% uop: the name of the operator
% op: a string defining the computation
% ztype: the type of z for z=f(x)
% xtype: the type of x for z=f(x)
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2025, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
f = fopen ('control.m4', 'w') ;
fprintf (f, 'm4_divert(-1)\n') ;
[zname, zunsigned, zbits] = codegen_type (ztype) ;
[xname, xunsigned, xbits] = codegen_type (xtype) ;
name = sprintf ('%s_%s_%s', uop, zname, xname) ;
% determine if the op is identity with no typecast
is_identity = isequal (uop, 'identity') ;
no_typecast = isequal (ztype, xtype) ;
if (is_identity && no_typecast)
% disable the _uop_apply method
fprintf (f, 'm4_define(`_uop_apply'', `_uop_apply__(none)'')\n') ;
fprintf (f, 'm4_define(`if_uop_apply_enabled'', `-1'')\n') ;
else
fprintf (f, 'm4_define(`_uop_apply'', `_uop_apply__%s'')\n', name) ;
fprintf (f, 'm4_define(`if_uop_apply_enabled'', `0'')\n') ;
end
if (is_identity)
% identity ops are never disabled
fprintf (f, 'm4_define(`GB_type_enabled'', `#define GB_TYPE_ENABLED 1'')\n');
else
codegen_type_enabled (f, xname) ;
end
% function names
fprintf (f, 'm4_define(`_uop_tran'', `_uop_tran__%s'')\n', name) ;
% type of C and A
fprintf (f, 'm4_define(`GB_ctype'', `#define GB_C_TYPE %s'')\n', ztype) ;
fprintf (f, 'm4_define(`GB_atype'', `#define GB_A_TYPE %s'')\n', xtype) ;
fprintf (f, 'm4_define(`GB_xtype'', `#define GB_X_TYPE %s'')\n', xtype) ;
fprintf (f, 'm4_define(`GB_ztype'', `#define GB_Z_TYPE %s'')\n', ztype) ;
A_is_pattern = (isempty (strfind (op, 'xarg'))) ;
% to get an entry from A
if (A_is_pattern)
% A(i,j) is not needed
fprintf (f, 'm4_define(`GB_declarea'', `#define GB_DECLAREA(aij)'')\n') ;
fprintf (f, 'm4_define(`GB_geta'', `#define GB_GETA(aij,Ax,pA,A_iso)'')\n') ;
else
% A is never iso
fprintf (f, 'm4_define(`GB_declarea'', `#define GB_DECLAREA(aij) %s aij'')\n', xtype) ;
fprintf (f, 'm4_define(`GB_geta'', `#define GB_GETA(aij,Ax,pA,A_iso) aij = Ax [pA]'')\n') ;
end
% type-specific iminv
if (~isempty (strfind (op, 'iminv')))
if (zunsigned)
op = strrep (op, 'iminv (', sprintf ('idiv_uint%d (1, ', zbits)) ;
else
op = strrep (op, 'iminv (', sprintf ('idiv_int%d (1, ', zbits)) ;
end
end
% create the unary operator
op = strrep (op, 'xarg', 'x') ;
fprintf (f, 'm4_define(`GB_unaryop'', `#define GB_UNARYOP(z,x) z = %s'')\n', op) ;
% create the disable flag
disable = sprintf ('defined(GxB_NO_%s)', upper (uop)) ;
disable = [disable (sprintf (' || defined(GxB_NO_%s)', upper (zname)))] ;
if (~isequal (zname, xname))
disable = [disable (sprintf (' || defined(GxB_NO_%s)', upper (xname)))] ;
end
fprintf (f, 'm4_define(`GB_disable'', `#if (%s)\n#define GB_DISABLE 1\n#else\n#define GB_DISABLE 0\n#endif\n'')\n', disable) ;
fprintf (f, 'm4_divert(0)\n') ;
fclose (f) ;
% construct the *.c file
cmd = sprintf ('cat control.m4 Generator/GB_uop.c | m4 -P | awk -f codegen_blank.awk > ../../FactoryKernels/GB_uop__%s.c', name) ;
fprintf ('.') ;
system (cmd) ;
% append to the *.h file
system ('cat control.m4 Generator/GB_uop.h | m4 -P | awk -f codegen_blank.awk | grep -v SPDX >> ../../FactoryKernels/GB_uop__include.h') ;
delete ('control.m4') ;
|
