-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathTrmvp_un_unb_var1.m
85 lines (64 loc) · 2.55 KB
/
Trmvp_un_unb_var1.m
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
% Copyright 2023 The University of Texas at Austin
%
% For licensing information see
% http://www.cs.utexas.edu/users/flame/license.html
%
% Programmed by: Anni Liu
% Interpret the function: http://edx-org-utaustinx.s3.amazonaws.com/UT501x/PictureFLAME/PictureFLAME.html
function [ y_out ] = Trmvp_un_unb_var1( U, x, y )
[ UTL, UTR, ...
UBL, UBR ] = FLA_Part_2x2( U, ...
0, 0, 'FLA_TL' );
[ xT, ...
xB ] = FLA_Part_2x1( x, ...
0, 'FLA_TOP' );
[ yT, ...
yB ] = FLA_Part_2x1( y, ...
0, 'FLA_TOP' );
while ( size( UTL, 1 ) < size( U, 1 ) )
[ U00, u01, U02, ...
u10t, upsilon11, u12t, ...
U20, u21, U22 ] = FLA_Repart_2x2_to_3x3( UTL, UTR, ...
UBL, UBR, ...
1, 1, 'FLA_BR' );
[ x0, ...
chi1, ...
x2 ] = FLA_Repart_2x1_to_3x1( xT, ...
xB, ...
1, 'FLA_BOTTOM' );
[ y0, ...
psi1, ...
y2 ] = FLA_Repart_2x1_to_3x1( yT, ...
yB, ...
1, 'FLA_BOTTOM' );
%------------------------------------------------------------%
psi1 = upsilon11 * chi1 + laff_dot(u12t, x2) + psi1;
%------------------------------------------------------------%
[ UTL, UTR, ...
UBL, UBR ] = FLA_Cont_with_3x3_to_2x2( U00, u01, U02, ...
u10t, upsilon11, u12t, ...
U20, u21, U22, ...
'FLA_TL' );
[ xT, ...
xB ] = FLA_Cont_with_3x1_to_2x1( x0, ...
chi1, ...
x2, ...
'FLA_TOP' );
[ yT, ...
yB ] = FLA_Cont_with_3x1_to_2x1( y0, ...
psi1, ...
y2, ...
'FLA_TOP' );
end
y_out = [ yT
yB ];
return
% Test the function
A = randi( [ -2, 2 ], 4, 4 )
x = randi( [ -2, 2 ], 4, 1 )
y = randi( [ -2, 2 ], 4, 1 )
if ( isequal( Trmvp_un_unb_var1( A, x, y ), triu( A ) * x + y ) )
disp( 'Trmvp_un_unb_var1 appears to be correct' )
else
disp( 'Trmvp_un_unb_var1 has a problem' )
end