def GaussianElimination_unb(A):
"""
GaussianElimination_unb(matrix)
Computes the Gauss Transform of the input matrix.
Traverses matrix A from TOP-LEFT to BOTTOM-RIGHT.
"""
ATL, ATR, \
ABL, ABR = flame.part_2x2(A, \
0, 0, 'TL')
while ATL.shape[0] < A.shape[0]:
A00, a01, A02, \
a10t, alpha11, a12t, \
A20, a21, A22 = flame.repart_2x2_to_3x3(ATL, ATR, \
ABL, ABR, \
1, 1, 'BR')
laff.invscal(alpha11, a21) # a21 := a21 / alpha11
laff.ger(-1.0, a21, a12t, A22) # A22 := A22 - a21 * a12t
ATL, ATR, \
ABL, ABR = flame.cont_with_3x3_to_2x2(A00, a01, A02, \
a10t, alpha11, a12t, \
A20, a21, A22, \
'TL')
flame.merge_2x2(ATL, ATR, \
ABL, ABR, A)
def GaussianElimination_unb(A):
"""
GaussianElimination_unb(matrix)
Computes the Gauss Transform of the input matrix.
Traverses matrix A from TOP-LEFT to BOTTOM-RIGHT.
"""
ATL, ATR, \
ABL, ABR = flame.part_2x2(A, \
0, 0, 'TL')
while ATL.shape[0] < A.shape[0]:
A00, a01, A02, \
a10t, alpha11, a12t, \
A20, a21, A22 = flame.repart_2x2_to_3x3(ATL, ATR, \
ABL, ABR, \
1, 1, 'BR')
laff.invscal( alpha11, a21 ) # a21 := a21 / alpha11
laff.ger( -1.0, a21, a12t, A22 ) # A22 := A22 - a21 * a12t
ATL, ATR, \
ABL, ABR = flame.cont_with_3x3_to_2x2(A00, a01, A02, \
a10t, alpha11, a12t, \
A20, a21, A22, \
'TL')
flame.merge_2x2(ATL, ATR, \
ABL, ABR, A)
def trsv_utn( U, b ):
UTL, UTR, \
UBL, UBR = flame.part_2x2(U, \
0, 0, 'TL')
bT, \
bB = flame.part_2x1(b, \
0, 'TOP')
while UTL.shape[0] < U.shape[0]:
U00, u01, U02, \
u10t, upsilon11, u12t, \
U20, u21, U22 = flame.repart_2x2_to_3x3(UTL, UTR, \
UBL, UBR, \
1, 1, 'BR')
b0, \
beta1, \
b2 = flame.repart_2x1_to_3x1(bT, \
bB, \
1, 'BOTTOM')
#------------------------------------------------------------#
invscal( upsilon11, beta1 )
axpy( -beta1, u12t, b2 )
#------------------------------------------------------------#
UTL, UTR, \
UBL, UBR = flame.cont_with_3x3_to_2x2(U00, u01, U02, \
u10t, upsilon11, u12t, \
U20, u21, U22, \
'TL')
bT, \
bB = flame.cont_with_3x1_to_2x1(b0, \
beta1, \
b2, \
'TOP')
flame.merge_2x1(bT, \
bB, b)
