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
111
112
113
114
115
116
117
|
function test25
%TEST25 test GxB_select
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
fprintf ('\ntest25: GxB_select tests\n') ;
[~, ~, ~, types, ~, select_ops] = GB_spec_opsall ;
types = types.all ;
rng ('default') ;
m = 10 ;
n = 6 ;
dt = struct ('inp0', 'tran') ;
for k1 = 1:length(types)
atype = types {k1} ;
fprintf ('%s: ', atype) ;
for A_is_hyper = 0:1
for A_is_csc = 0:1
for C_is_hyper = 0:1
for C_is_csc = 0:1
for M_is_hyper = 0:1
for M_is_csc = 0:1
if (A_is_hyper)
ha = 1 ;
else
ha = 0 ;
end
if (C_is_hyper)
hc = 1 ;
else
hc = 0 ;
end
if (M_is_hyper)
hm = 1 ;
else
hm = 0 ;
end
A = GB_spec_random (m, n, 0.3, 100, atype, A_is_csc, A_is_hyper, ha) ;
A.matrix (:,1) = rand (m,1) ;
A.pattern (:,1) = true (m,1) ;
Cin = GB_spec_random (m, n, 0.3, 100, atype, C_is_csc, C_is_hyper, hc) ;
B = GB_spec_random (n, m, 0.3, 100, atype, A_is_csc, A_is_hyper, ha) ;
cin = GB_mex_cast (0, atype) ;
% Mask = (sprand (m, n, 0.5) ~= 0) ;
Mask = GB_random_mask (m, n, 0.5, M_is_csc, M_is_hyper) ;
Mask.hyper_switch = hm ;
fprintf ('.') ;
for k2 = 1:length(select_ops)
op = select_ops {k2} ;
% fprintf ('%s ', op) ;
for k = -m:3:n % Was: [-m:n]
% no mask
C1 = GB_spec_select (Cin, [], [], op, A, k, []) ;
C2 = GB_mex_select (Cin, [], [], op, A, k, [], 'test') ;
GB_spec_compare (C1, C2) ;
% no mask, with accum
C1 = GB_spec_select (Cin, [], 'plus', op, A, k, []) ;
C2 = GB_mex_select (Cin, [], 'plus', op, A, k, [], 'test') ;
GB_spec_compare (C1, C2) ;
% with mask
C1 = GB_spec_select (Cin, Mask, [], op, A, k, []) ;
C2 = GB_mex_select (Cin, Mask, [], op, A, k, [], 'test') ;
GB_spec_compare (C1, C2) ;
% with mask and accum
C1 = GB_spec_select (Cin, Mask, 'plus', op, A, k, []) ;
C2 = GB_mex_select (Cin, Mask, 'plus', op, A, k, [], 'test') ;
GB_spec_compare (C1, C2) ;
% no mask, transpose
C1 = GB_spec_select (Cin, [], [], op, B, k, dt) ;
C2 = GB_mex_select (Cin, [], [], op, B, k, dt, 'test') ;
GB_spec_compare (C1, C2) ;
% no mask, with accum, transpose
C1 = GB_spec_select (Cin, [], 'plus', op, B, k, dt) ;
C2 = GB_mex_select (Cin, [], 'plus', op, B, k, dt, 'test') ;
GB_spec_compare (C1, C2) ;
% with mask, transpose
C1 = GB_spec_select (Cin, Mask, [], op, B, k, dt) ;
C2 = GB_mex_select (Cin, Mask, [], op, B, k, dt, 'test') ;
GB_spec_compare (C1, C2) ;
% with mask and accum, transpose
C1 = GB_spec_select (Cin, Mask, 'plus', op, B, k, dt) ;
C2 = GB_mex_select (Cin, Mask, 'plus', op, B, k, dt, 'test') ;
GB_spec_compare (C1, C2) ;
end
end
end
end
end
end
end
end
fprintf ('\n') ;
end
fprintf ('\ntest25: all tests passed\n') ;
|
