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
|
function gbtest82
%GBTEST82 test complex A*B, A'*B, A*B', A'*B', A+B
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
rng ('default') ;
nlist = [1 4 10] ;
r = complex ([-1 1]) ;
maxerr = 0 ;
for m = nlist
for n = nlist
for k = nlist
A = GrB.random (k, m, (m*5)/(k*m), 'range', r) ;
B = GrB.random (k, n, (n*5)/(k*n), 'range', r) ;
C = double (A).'*double (B) ;
C2 = GrB.mxm (A, '+.*', B, struct ('in0', 'transpose')) ;
err = norm (C-C2,1) ;
maxerr = max (maxerr, err) ;
assert (err < 1e-12)
end
end
end
fprintf ('All complex A''*B tests passed, maxerr %g\n', maxerr) ;
maxerr = 0 ;
for m = nlist
for n = nlist
A = GrB.random (m, n, (m*5)/(k*m), 'range', r) ;
B = GrB.random (m, n, (n*5)/(k*n), 'range', r) ;
C = double (A) + double (B) ;
C2 = A + B ;
err = norm (C-C2,1) ;
maxerr = max (maxerr, err) ;
assert (err < 1e-12)
end
end
fprintf ('All complex A+B tests passed, maxerr %g\n', maxerr) ;
maxerr = 0 ;
for m = nlist
for n = nlist
for k = nlist
for at = 0:1
for bt = 0:1
if (at)
A = GrB.random (k, m, (n*5)/(k*m), 'range', r) ;
else
A = GrB.random (m, k, (m*5)/(k*m), 'range', r) ;
end
if (bt)
B = GrB.random (n, k, (m*5)/(k*m), 'range', r) ;
else
B = GrB.random (k, n, (m*5)/(k*m), 'range', r) ;
end
desc = struct ;
if (at)
desc.in0 = 'transpose' ;
end
if (bt)
desc.in1 = 'transpose' ;
end
M = sparse (m, n) ;
M (1,1) = 1 ; %#ok
C = GrB.mxm (A, '+.*', B, desc) ;
Cin = GrB (m, n, 'double complex') ;
CM = GrB.mxm (Cin, M, A, '+.*', B, desc) ;
A = double (A) ;
B = double (B) ;
if (at)
if (bt)
C2 = A.'*B.' ;
CM2 = (A.'*B.') .* M ;
else
C2 = A.'*B ;
CM2 = (A.'*B) .* M ;
end
else
if (bt)
C2 = A*B.' ;
CM2 = (A*B.') .* M ;
else
C2 = A*B ;
CM2 = (A*B) .* M ;
end
end
err = norm (C-C2,1) ;
maxerr = max (maxerr, err) ;
assert (err < 1e-12)
err = norm (CM-CM2,1) ;
maxerr = max (maxerr, err) ;
assert (err < 1e-12)
end
end
end
end
end
fprintf ('maxerr: %g\n', maxerr) ;
fprintf ('gbtest82: all tests passed\n') ;
|
