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
|
function C = gb_bitwise (op, A, B, assumedtype)
%GB_BITWISE bitwise AND, OR, XOR, ...
% SuiteSparse:GraphBLAS, Timothy A. Davis, (c) 2017-2022, All Rights Reserved.
% SPDX-License-Identifier: Apache-2.0
if (isobject (A))
A = A.opaque ;
end
if (isobject (B))
B = B.opaque ;
end
atype = gbtype (A) ;
btype = gbtype (B) ;
if (gb_contains (atype, 'complex') || gb_contains (btype, 'complex'))
error ('GrB:error', 'inputs must be real') ;
end
if (isequal (atype, 'logical') || isequal (btype, 'logical'))
error ('GrB:error', 'inputs must not be logical') ;
end
if (~gb_contains (assumedtype, 'int'))
error ('GrB:error', 'assumedtype must be an integer type') ;
end
% C will have the same type as A on input
ctype = atype ;
if (isequal (atype, 'double') || isequal (atype, 'single'))
A = gbnew (A, assumedtype) ;
atype = assumedtype ;
end
if (isequal (op, 'bitshift'))
if (~isequal (btype, 'int8'))
% convert B to int8, and ensure all values are in range -64:64
% ensure all entries in B are <= 64
B = gbapply2 (['min.' btype], B, 64) ;
if (gb_issigned (btype))
% ensure all entries in B are >= -64
B = gbapply2 (['max.' btype], B, -64) ;
end
B = gbnew (B, 'int8') ;
end
a_is_scalar = gb_isscalar (A) ;
b_is_scalar = gb_isscalar (B) ;
if (a_is_scalar && ~b_is_scalar)
% A is a scalar, B is a matrix
C = gbapply2 (['bitshift.' atype], gbfull (A), B) ;
elseif (~a_is_scalar && b_is_scalar)
% A is a matrix, B is a scalar
C = gbapply2 (['bitshift.' atype], A, gbfull (B)) ;
else
% both A and B are matrices, or both are scalars
% expand B by padding it with zeros from the pattern of A
B = gbeadd ('1st.int8', B, gb_expand (0, A, 'int8')) ;
C = gbemult (['bitshift.' atype], A, B) ;
end
else
if (isequal (btype, 'double') || isequal (btype, 'single'))
B = gbnew (B, assumedtype) ;
btype = assumedtype ;
end
if (~isequal (atype, btype))
error ('GrB:error', 'integer inputs must have the same type') ;
end
switch (op)
case { 'bitxor', 'bitor' }
C = gb_eadd (A, op, B) ;
case { 'bitand' }
C = gb_emult (A, op, B) ;
end
end
if (~isequal (gbtype (C), ctype))
C = gbnew (C, ctype) ;
end
|
