def trsm_rutu(U, B):
BT, \
BB = flame.part_2x1(B, \
0, 'TOP')
while BT.shape[0] < B.shape[0]:
B0, \
b1t, \
B2 = flame.repart_2x1_to_3x1(BT, \
BB, \
1, 'BOTTOM')
#------------------------------------------------------------#
#TODO: Fix trsv.py to handle rows properly
#Sorry for the hackiness in the fix here, I got pressed for time
trsv( 'Upper triangular', 'No transpose', 'Unit diagonal', U, transpose( b1t ) )
#------------------------------------------------------------#
BT, \
BB = flame.cont_with_3x1_to_2x1(B0, \
b1t, \
B2, \
'TOP')
flame.merge_2x1(BT, \
BB, B)
def trsm_rlnn(L, B):
BT, \
BB = flame.part_2x1(B, \
0, 'TOP')
while BT.shape[0] < B.shape[0]:
B0, \
b1t, \
B2 = flame.repart_2x1_to_3x1(BT, \
BB, \
1, 'BOTTOM')
#------------------------------------------------------------#
#TODO: Fix trsv.py to handle rows properly
#Sorry for the hackiness in the fix here, I got pressed for time
trsv('Lower triangular', 'Transpose', 'Nonunit diagonal', L,
transpose(b1t))
#------------------------------------------------------------#
BT, \
BB = flame.cont_with_3x1_to_2x1(B0, \
b1t, \
B2, \
'TOP')
flame.merge_2x1(BT, \
BB, B)
def trsm_lutu(U, B):
BL, BR = flame.part_1x2(B, \
0, 'LEFT')
while BL.shape[1] < B.shape[1]:
B0, b1, B2 = flame.repart_1x2_to_1x3(BL, BR, \
1, 'RIGHT')
#------------------------------------------------------------#
trsv('Upper triangular', 'Transpose', 'Unit diagonal', U, b1)
#------------------------------------------------------------#
BL, BR = flame.cont_with_1x3_to_1x2(B0, b1, B2, \
'LEFT')
flame.merge_1x2(BL, BR, B)
def trsm_llnu(L, B):
BL, BR = flame.part_1x2(B, \
0, 'LEFT')
while BL.shape[1] < B.shape[1]:
B0, b1, B2 = flame.repart_1x2_to_1x3(BL, BR, \
1, 'RIGHT')
#------------------------------------------------------------#
trsv( 'Lower triangular', 'No transpose', 'Unit diagonal', L, b1 )
#------------------------------------------------------------#
BL, BR = flame.cont_with_1x3_to_1x2(B0, b1, B2, \
'LEFT')
flame.merge_1x2(BL, BR, B)
def trsm_llnn(L, B):
BL, BR = flame.part_1x2(B, \
0, 'LEFT')
while BL.shape[1] < B.shape[1]:
B0, b1, B2 = flame.repart_1x2_to_1x3(BL, BR, \
1, 'RIGHT')
#------------------------------------------------------------#
trsv( 'Lower triangular', 'No transpose', 'Nonunit diagonal', L, b1 )
#------------------------------------------------------------#
BL, BR = flame.cont_with_1x3_to_1x2(B0, b1, B2, \
'LEFT')
flame.merge_1x2(BL, BR, B)
def chol_unb(A):
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')
#------------------------------------------------------------#
# alpha11[0,0] = np.sqrt( alpha11[0,0] )
# laff.invscal( alpha11, a21 )
# laff.ger( -1, a21, a21, A22 ) #Not tril
# laff.dots( a10t, a10t, alpha11 )
# alpha11[0,0] = np.sqrt( alpha11[0,0] )
# a21 -= A20 * np.transpose(a10t) #Not laff calls
# laff.invscal( alpha11, a21 )
trsv('Lower triangular', 'Nonunit diagonal', 'No transpose', A00, a10t)
dots(a10t, a10t, alpha11)
alpha11[0, 0] = np.sqrt(alpha11[0, 0])
#------------------------------------------------------------#
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 chol_unb(A):
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')
#------------------------------------------------------------#
# alpha11[0,0] = np.sqrt( alpha11[0,0] )
# laff.invscal( alpha11, a21 )
# laff.ger( -1, a21, a21, A22 ) #Not tril
# laff.dots( a10t, a10t, alpha11 )
# alpha11[0,0] = np.sqrt( alpha11[0,0] )
# a21 -= A20 * np.transpose(a10t) #Not laff calls
# laff.invscal( alpha11, a21 )
trsv( 'Lower triangular', 'Nonunit diagonal', 'No transpose', A00, a10t )
dots( a10t, a10t, alpha11 )
alpha11[0,0] = np.sqrt( alpha11[0,0] )
#------------------------------------------------------------#
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)
