some more fixes

Author: lkdvos

src/algorithms/derivatives/mpo_derivatives.jl

@@ -169,16 +169,15 @@ function (H::PrecomputedDerivative)(x::AbstractTensorMap)
     R_fused = fuse_legs(H.rightenv, 0, numin(x))
     x_fused = fuse_legs(x, numout(x), numin(x))
 
-
     TC = TensorOperations.promote_contract(scalartype(x_fused), scalartype(R_fused))
     xR = TensorOperations.tensoralloc_contract(
-        TC, x_fused, ((1,), (2,)), false, R_fused, ((1,), (2, 3)), false, ((1, 2), (3,)), Val(true), H.allocator
+        TC, x_fused, ((1,), (2,)), false, R_fused, ((1,), (2, 3)), false, ((1, 2), (3,)), Val(true), allocator
     )
 
-    mul_front!(xR, x_fused, R_fused, One(), Zero(), H.backend, H.allocator)
+    mul_front!(xR, x_fused, R_fused, One(), Zero(), H.backend, allocator)
 
     LxR = H.leftenv * xR
-    TensorOperations.tensorfree!(xR, H.allocator)
+    TensorOperations.tensorfree!(xR, allocator)
 
     reset!(allocator, cp)
     return TensorMap{scalartype(LxR)}(LxR.data, codomain(H.leftenv) ← domain(H.rightenv))

src/utility/utility.jl

@@ -275,8 +275,9 @@ function mul_front!(
             α, β, backend, allocator
         )
     end
+    reset!(allocator, cp)
 
-    return reset!(allocator, cp)
+    return C
 end
 
 function mul_tail(
@@ -324,7 +325,8 @@ function mul_tail!(
         )
     end
 
-    return reset!(allocator, cp)
+    reset!(allocator, cp)
+    return C
 end
 
 @inline fuse_legs(x::TensorMap, N₁::Int, N₂::Int) = fuse_legs(x, Val(N₁), Val(N₂))