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
|
function codegen_unop_method (unop, op, fcast, ztype, xtype)
%CODEGEN_UNOP_METHOD create a function to compute C=unop(cast(A))
%
% codegen_unop_method (unop, op, fcast, ztype, xtype)
%
% unop: 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-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
f = fopen ('control.m4', 'w') ;
[zname, zunsigned, zbits] = codegen_type (ztype) ;
[xname, xunsigned, xbits] = codegen_type (xtype) ;
name = sprintf ('%s_%s_%s', unop, zname, xname) ;
% determine if the op is identity with no typecast
is_identity = isequal (unop, 'identity') ;
no_typecast = isequal (ztype, xtype) ;
if (is_identity && no_typecast)
fprintf (f, 'define(`_unop_apply'', `_unop_apply__(none)'')\n') ;
fprintf (f, 'define(`if_unop_apply_enabled'', `#if 0'')\n') ;
fprintf (f, 'define(`endif_unop_apply_enabled'', `#endif'')\n') ;
else
fprintf (f, 'define(`_unop_apply'', `_unop_apply__%s'')\n', name) ;
fprintf (f, 'define(`if_unop_apply_enabled'', `'')\n') ;
fprintf (f, 'define(`endif_unop_apply_enabled'', `'')\n') ;
end
% function names
fprintf (f, 'define(`_unop_tran'', `_unop_tran__%s'')\n', name) ;
% type of C and A
fprintf (f, 'define(`GB_ctype'', `%s'')\n', ztype) ;
fprintf (f, 'define(`GB_atype'', `%s'')\n', xtype) ;
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, 'define(`GB_geta'', `;'')\n') ;
else
% A is not iso, so GBX (Ax, p, A->iso) is not needed
fprintf (f, 'define(`GB_geta'', `%s $1 = $2 [$3]'')\n', xtype) ;
end
% type-specific IMINV
if (~isempty (strfind (op, 'IMINV')))
if (zunsigned)
op = strrep (op, 'IMINV', 'IMINV_UNSIGNED') ;
else
op = strrep (op, 'IMINV', 'IMINV_SIGNED') ;
end
op = strrep (op, ')', sprintf (', %d)', zbits)) ;
end
% create the unary operator
op = strrep (op, 'xarg', '`$2''') ;
fprintf (f, 'define(`GB_unaryop'', `$1 = %s'')\n', op) ;
% create the cast operator
if (A_is_pattern)
% cast (A(i,j)) is not needed
fprintf (f, 'define(`GB_cast'', `;'')\n') ;
else
fcast = strrep (fcast, 'zarg', '`$1''') ;
fcast = strrep (fcast, 'xarg', '`$2''') ;
fprintf (f, 'define(`GB_cast'', `%s'')\n', fcast) ;
end
% create the disable flag
disable = sprintf ('GxB_NO_%s', upper (unop)) ;
disable = [disable (sprintf (' || GxB_NO_%s', upper (zname)))] ;
if (~isequal (zname, xname))
disable = [disable (sprintf (' || GxB_NO_%s', upper (xname)))] ;
end
fprintf (f, 'define(`GB_disable'', `(%s)'')\n', disable) ;
fclose (f) ;
% construct the *.c file
cmd = sprintf (...
'cat control.m4 Generator/GB_unop.c | m4 | tail -n +11 > Generated2/GB_unop__%s.c', ...
name) ;
fprintf ('.') ;
system (cmd) ;
% append to the *.h file
cmd = sprintf (...
'cat control.m4 Generator/GB_unop.h | m4 | tail -n +11 >> Generated2/GB_unop__include.h') ;
system (cmd) ;
delete ('control.m4') ;
|
