adapt pullbacks

Author: lkdvos

src/tensors/factorizations/factorizations.jl

@@ -42,7 +42,8 @@ import MatrixAlgebraKit: default_algorithm,
                          left_orth!, right_orth!, left_null!, right_null!,
                          truncate, findtruncated, findtruncated_svd,
                          diagview, isisometry
-using MatrixAlgebraKit: qr_compact_pullback!, lq_compact_pullback!,
+using MatrixAlgebraKit: qr_pullback!, qr_null_pullback!,
+                        lq_pullback!, lq_null_pullback!,
                         svd_pullback!, svd_trunc_pullback!,
                         eig_pullback!, eig_trunc_pullback!,
                         eigh_pullback!, eigh_trunc_pullback!,

src/tensors/factorizations/pullbacks.jl

@@ -1,4 +1,4 @@
-for pullback! in (:qr_compact_pullback!, :lq_compact_pullback!,
+for pullback! in (:qr_pullback!, :lq_pullback!,
                   :left_polar_pullback!, :right_polar_pullback!)
     @eval function MatrixAlgebraKit.$pullback!(Δt::AbstractTensorMap, t::AbstractTensorMap,
                                                F, ΔF; kwargs...)
@@ -10,6 +10,17 @@ for pullback! in (:qr_compact_pullback!, :lq_compact_pullback!,
         return Δt
     end
 end
+for pullback! in (:qr_null_pullback!, :lq_null_pullback!)
+    @eval function MatrixAlgebraKit.$pullback!(Δt::AbstractTensorMap, t::AbstractTensorMap,
+                                               F, ΔF; kwargs...)
+        foreachblock(Δt, t) do c, (Δb, b)
+            Fc = block(F, c)
+            ΔFc = block(ΔF, c)
+            return $pullback!(Δb, b, Fc, ΔFc; kwargs...)
+        end
+        return Δt
+    end
+end
 
 _notrunc_ind(t) = SectorDict(c => Colon() for c in blocksectors(t))