@@ -6,13 +6,15 @@ import BlockArrays: BlockSlice, block, blockindex, blockvec
66import BlockBandedMatrices: subblockbandwidths, blockbandwidths, AbstractBandedBlockBandedLayout, AbstractBandedBlockBandedMatrix
77import Base: size, axes, getindex, + , - , * , / , == , \ , OneTo, oneto, replace_in_print_matrix, copy, diff, getproperty, adjoint, transpose, tail, _sum, inv, show, summary
88import LazyArrays: paddeddata, AbstractLazyLayout
9- import LazyBandedMatrices: BlockBroadcastMatrix, BlockVec, BandedLazyLayouts, AbstractLazyBandedBlockBandedLayout, UpperOrLowerTriangular
10- import LinearAlgebra: BlasInt, eigvals
9+ import LazyBandedMatrices: BlockBroadcastMatrix, BlockVec, AbstractLazyBandedBlockBandedLayout
10+ import LinearAlgebra: BlasInt, eigvals, UpperOrLowerTriangular
1111import MatrixFactorizations: reversecholcopy
1212import FillArrays: AbstractFill
1313import FillArrays: SquareEye
1414import InfiniteArrays: OneToInf
1515
16+ import LazyBandedMatrices. LazyArraysBandedMatricesExt: BandedLazyLayouts
17+
1618
1719export BBBArrowheadMatrix
1820
186188
187189function materialize! (M:: MatMulVecAdd{<:ArrowheadLayouts,<:AbstractStridedLayout,<:AbstractStridedLayout} )
188190 α, A, x_in, β, y_in = M. α, M. A, M. B, M. β, M. C
189- x = PseudoBlockArray (x_in, (axes (A,2 ), ))
190- y = PseudoBlockArray (y_in, (axes (A,1 ),))
191+ x = BlockedArray (x_in, (axes (A,2 ), ))
192+ y = BlockedArray (y_in, (axes (A,1 ),))
191193 m,n = size (A. A)
192194
193195 _fill_lmul! (β, y)
209211
210212function materialize! (M:: MatMulMatAdd{<:ArrowheadLayouts,<:AbstractColumnMajor,<:AbstractColumnMajor} )
211213 α, A, X_in, β, Y_in = M. α, M. A, M. B, M. β, M. C
212- X = PseudoBlockArray (X_in, (axes (A,2 ), axes (X_in,2 )))
213- Y = PseudoBlockArray (Y_in, (axes (A,1 ), axes (X_in,2 )))
214+ X = BlockedArray (X_in, (axes (A,2 ), axes (X_in,2 )))
215+ Y = BlockedArray (Y_in, (axes (A,1 ), axes (X_in,2 )))
214216 m,n = size (A. A)
215217
216218 _fill_lmul! (β, Y)
232234
233235function materialize! (M:: MatMulMatAdd{<:AbstractColumnMajor,<:ArrowheadLayouts,<:AbstractColumnMajor} )
234236 α, X_in, A, β, Y_in = M. α, M. A, M. B, M. β, M. C
235- X = PseudoBlockArray (X_in, (axes (X_in,1 ), axes (A,1 )))
236- Y = PseudoBlockArray (Y_in, (axes (Y_in,1 ), axes (A,2 )))
237+ X = BlockedArray (X_in, (axes (X_in,1 ), axes (A,1 )))
238+ Y = BlockedArray (Y_in, (axes (Y_in,1 ), axes (A,2 )))
237239 m,n = size (A. A)
238240
239241 _fill_lmul! (β, Y)
@@ -439,7 +441,7 @@ for (UNIT, Tri) in (('U',UnitUpperTriangular), ('N', UpperTriangular))
439441 N = blocksize (P,1 )
440442
441443 # impose block structure
442- b = PseudoBlockArray (dest, (axes (P,1 ),))
444+ b = BlockedArray (dest, (axes (P,1 ),))
443445 b̃_1 = view (b, Block (1 ))
444446
445447 for K = 1 : min (N- 1 ,length (B))
@@ -463,7 +465,7 @@ for (UNIT, Tri) in (('U',UnitLowerTriangular), ('N', LowerTriangular))
463465 m = length (D)
464466
465467 # impose block structure
466- b = PseudoBlockArray (dest, (axes (P,1 ),))
468+ b = BlockedArray (dest, (axes (P,1 ),))
467469 b̃_1 = view (b, Block (1 ))
468470 ArrayLayouts. ldiv! ($ Tri (A), b̃_1)
469471
0 commit comments