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
|
function test246 (dohack)
%TEST246 test GrB_mxm with different kinds of parallelism
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
rng ('default') ;
n = 1000 ;
A = sprand (n, n, 0.5) ;
% save current global settings
save = GB_mex_hack ;
[nthreads_save chunk_save] = nthreads_get ;
% modify the global settings
hack = save ;
if (nargin < 1)
dohack = 2 ; % use 0 for default, 2 for prior setting in v7.2.0
end
hack (1) = dohack ; % modify "very_costly" in GxB_AxB_saxpy3_slice_balanced
GB_mex_hack (hack) ;
GrB.burble (0) ;
semiring.multiply = 'times' ;
semiring.add = 'plus' ;
semiring.class = 'double' ;
for k = [1 2 4 16 128]
S = sparse (n, k) ;
M = spones (sprand (n, k, 0.1)) ;
for da = [0.001 0.5]
A = sprand (n, n, da) ;
H.matrix = A ;
H.sparsity = 1 ;
for db = [0 0.01, 0.5]
if (db == 0 && k == 4)
B = sprand (n, k, 0.01) ;
B (:,2) = pi ;
B (1:(n/10),3) = 3 ;
elseif (db == 0 && k > 1)
B = sprand (n, k, 0.01) ;
B (:,2) = pi ;
else
B = sprand (n, k, db) ;
end
for method = [0 1 2]
if (method == 0)
desc = [ ] ;
elseif (method == 1)
desc.axb = 'gustavson' ;
else
desc.axb = 'hash' ;
end
for threads = [1 4 16]
nthreads_set (threads, 1) ;
% no mask
C1 = A*B ;
C2 = GB_mex_mxm (S, [ ], [ ], semiring, A, B, desc) ;
err = norm (C1 - C2.matrix, 1) / max (1, norm (C1, 1)) ;
assert (err < 1e-12)
% with the mask
C1 = M.*(A*B) ;
C2 = GB_mex_mxm (S, M, [ ], semiring, A, B, desc) ;
err = norm (C1 - C2.matrix, 1) / max (1, norm (C1, 1)) ;
assert (err < 1e-12)
% hypersparse case
C2 = GB_mex_mxm (S, M, [ ], semiring, H, B, desc) ;
err = norm (C1 - C2.matrix, 1) / max (1, norm (C1, 1)) ;
assert (err < 1e-12)
end
end
end
end
end
% restore global settings
GrB.burble (0) ;
nthreads_set (nthreads_save, chunk_save) ;
GB_mex_hack (save) ;
fprintf ('\ntest246: all tests passed\n') ;
|
