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
|
function C = gb_max2 (op, A, B)
%GB_MAX2 2-input max
% Implements C = max (A,B)
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
[am, an, atype] = gbsize (A) ;
[bm, bn, btype] = gbsize (B) ;
a_is_scalar = (am == 1) && (an == 1) ;
b_is_scalar = (bm == 1) && (bn == 1) ;
ctype = gboptype (atype, btype) ;
if (a_is_scalar)
if (b_is_scalar)
% both A and B are scalars. Result is also a scalar.
C = gbeunion (op, A, 0, B, 0) ;
else
% A is a scalar, B is a matrix
if (gb_scalar (A) > 0)
% since A > 0, the result is full
A = gb_scalar_to_full (bm, bn, ctype, gb_fmt (B), A) ;
C = gbeadd (A, op, B) ;
else
% since A <= 0, the result is sparse.
C = gbapply2 (gbfull (A), op, B) ;
end
end
else
if (b_is_scalar)
% A is a matrix, B is a scalar
if (gb_scalar (B) > 0)
% since B > 0, the result is full
B = gb_scalar_to_full (am, an, ctype, gb_fmt (A), B) ;
C = gbeadd (A, op, B) ;
else
% since B <= 0, the result is sparse.
C = gbapply2 (A, op, gbfull (B)) ;
end
else
% both A and B are matrices. Result is sparse.
C = gbeunion (op, A, 0, B, 0) ;
end
end
|