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
|
function test158
%TEST158 test colscale (A*D) and rowscale (D*B) with positional ops
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
rng ('default') ;
[binops, ~, ~, ~, ~, ~] = GB_spec_opsall ;
pos = binops.positional ;
pos {end+1} = 'times' ;
pos {end+1} = 'div' ;
n = 30 ;
A = GB_spec_random (n, n, 0.05, 256, 'int64') ;
D.matrix = speye (n) ;
D.class = 'int64' ;
D.pattern = logical (spones (D.matrix)) ;
dnn = struct ;
dtn = struct ('inp0', 'tran') ;
dnt = struct ('inp1', 'tran') ;
dtt = struct ('inp0', 'tran', 'inp1', 'tran') ;
Cin = sparse (n,n) ;
semiring.add = 'plus' ;
semiring.class = 'int64' ;
for c = 1:3
if (c == 1)
A.class = 'int32' ;
D.class = 'int32' ;
semiring.class = 'int32' ;
elseif (c == 2)
A.class = 'int64' ;
D.class = 'int64' ;
semiring.class = 'int64' ;
else
A.class = 'int32' ;
D.class = 'int64' ;
semiring.class = 'int64' ;
end
fprintf ('\ntypes: %s %s %s\n', D.class, A.class, semiring.class) ;
for k = 1:length(pos)
op = pos {k} ;
fprintf ('op: %s\n', op) ;
semiring.multiply = op ;
% colscale: C = A*D
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, A, D, [ ]) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, A, D, [ ]) ;
GB_spec_compare (C1, C2) ;
% rowscale: C = D*A
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, D, A, [ ]) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, D, A, [ ]) ;
GB_spec_compare (C1, C2) ;
% colscale: C = A'*D
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, A, D, dtn) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, A, D, dtn) ;
GB_spec_compare (C1, C2) ;
% colscale: C = D'*A
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, D, A, dtn) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, D, A, dtn) ;
GB_spec_compare (C1, C2) ;
% rowscale: C = D*A'
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, D, A, dnt) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, D, A, dnt) ;
GB_spec_compare (C1, C2) ;
% rowscale: C = A*D'
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, A, D, dnt) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, A, D, dnt) ;
GB_spec_compare (C1, C2) ;
% colscale: C = A'*D'
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, A, D, dtt) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, A, D, dtt) ;
GB_spec_compare (C1, C2) ;
% rowscale: C = D'*B'
C1 = GB_spec_mxm (Cin, [ ], [ ], semiring, D, A, dtt) ;
C2 = GB_mex_mxm (Cin, [ ], [ ], semiring, D, A, dtt) ;
GB_spec_compare (C1, C2) ;
end
end
fprintf ('\ntest158: all tests passed\n') ;
|