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
|
function test145
%TEST145 test dot4
% GB_AxB_dot4 computes C+=A'*B when C is dense.
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
fprintf ('test145 -------------------- C+=A''*B when C is dense, with dot4\n') ;
GrB.burble (0) ;
rng ('default') ;
A = sparse (rand (4)) ;
B = sparse (rand (4)) ;
C = sparse (rand (4)) ;
AT = A' ;
BT = B' ;
semiring.add = 'plus' ;
semiring.multiply = 'div' ;
semiring.class = 'double' ;
[mult_op add_op id] = GB_spec_semiring (semiring) ;
dnn = struct ('axb', 'dot') ;
dtn = struct ('axb', 'dot', 'inp0', 'tran') ;
dnt = struct ('axb', 'dot', 'inp1', 'tran') ;
dtt = struct ('axb', 'dot', 'inp0', 'tran', 'inp1', 'tran') ;
C2 = GB_mex_mxm (C, [ ], add_op, semiring, A, B, dnn) ;
C1 = GB_spec_mxm (C, [ ], add_op, semiring, A, B, dnn) ;
GB_spec_compare (C1, C2) ;
C2 = GB_mex_mxm (C, [ ], add_op, semiring, AT, B, dtn) ;
C1 = GB_spec_mxm (C, [ ], add_op, semiring, AT, B, dtn) ;
GB_spec_compare (C1, C2) ;
C2 = GB_mex_mxm (C, [ ], add_op, semiring, A, BT, dnt) ;
C1 = GB_spec_mxm (C, [ ], add_op, semiring, A, BT, dnt) ;
GB_spec_compare (C1, C2) ;
C2 = GB_mex_mxm (C, [ ], add_op, semiring, AT, BT, dtt) ;
C1 = GB_spec_mxm (C, [ ], add_op, semiring, AT, BT, dtt) ;
GB_spec_compare (C1, C2) ;
X = 1./A ;
C1 = X*B ;
C2 = GB_mex_rdiv (A, B, 1003) ;
assert (norm (C1-C2,1) < 1e-5)
C2 = GB_mex_rdiv2 (A, B, false, false, 1003, 0) ;
assert (norm (C1-C2,1) < 1e-5)
C2 = GB_mex_rdiv2 (AT, B, true, false, 1003, 0) ;
assert (norm (C1-C2,1) < 1e-5)
C2 = GB_mex_rdiv2 (A, BT, false, true, 1003, 0) ;
assert (norm (C1-C2,1) < 1e-5)
C2 = GB_mex_rdiv2 (AT, BT, true, true, 1003, 0) ;
assert (norm (C1-C2,1) < 1e-5)
X = 1./B ;
C1 = A*X ;
C2 = GB_mex_rdiv2 (A, B, false, false, 1003, 1) ;
assert (norm (C1-C2,1) < 1e-5)
C2 = GB_mex_rdiv2 (AT, B, true, false, 1003, 1) ;
assert (norm (C1-C2,1) < 1e-5)
C2 = GB_mex_rdiv2 (A, BT, false, true, 1003, 1) ;
assert (norm (C1-C2,1) < 1e-5)
% update C in place with dot4:
X = 1./B ;
C0 = A*X ;
C1 = A*X + pi ;
[C2, inplace] = GB_mex_rdiv2 (A, B, false, false, 1003, 1, pi) ;
if (inplace)
assert (norm (C1-C2,1) < 1e-5)
else
assert (norm (C0-C2,1) < 1e-5)
end
[C2, inplace] = GB_mex_rdiv2 (AT, B, true, false, 1003, 1, pi) ;
if (inplace)
assert (norm (C1-C2,1) < 1e-5)
else
assert (norm (C0-C2,1) < 1e-5)
end
[C2, inplace] = GB_mex_rdiv2 (A, BT, false, true, 1003, 1, pi) ;
if (inplace)
assert (norm (C1-C2,1) < 1e-5)
else
assert (norm (C0-C2,1) < 1e-5)
end
fprintf ('test145: all tests passed\n') ;
|
