Update blockkron.jl

JuliaLinearAlgebra · Jul 23, 2024 · a369bf2 · a369bf2
1 parent cfb93f2
commit a369bf2
Showing 1 changed file with 8 additions and 1 deletion.
diff --git a/src/blockkron.jl b/src/blockkron.jl
@@ -125,8 +125,15 @@ function Base._maximum(f, a::DiagTrav, ::Colon; kws...)
     ret
 end
 
+pad(c, ax...) = PaddedArray(c, ax)
+
 function copy(M::Mul{<:LazyBlockBandedLayouts,<:DiagTravLayout{<:AbstractPaddedLayout}})
-    error("hi")
+    A,B = M.A, M.B
+
+    P = DiagTrav(paddeddata(B.array))
+    JR = blockaxes(P,1)
+    KR = blockcolsupport(A,JR)
+    pad(A[KR,JR] * P, axes(A,1))
 end
 
 struct InvDiagTrav{T, AA<:AbstractVector{T}} <: LayoutMatrix{T}