From a369bf2dde82ec9ddf55459ea91316cc81c0ea85 Mon Sep 17 00:00:00 2001
From: Sheehan Olver <solver@mac.com>
Date: Tue, 23 Jul 2024 16:38:03 +0100
Subject: [PATCH] Update blockkron.jl

---
 src/blockkron.jl | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

diff --git a/src/blockkron.jl b/src/blockkron.jl
index 2462b37..bf57ea7 100644
--- 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}