Skip to content

Commit deea721

Browse files
committed
Fix imports
1 parent b9a0173 commit deea721

2 files changed

Lines changed: 5 additions & 2 deletions

File tree

ext/TensorKitChainRulesCoreExt/TensorKitChainRulesCoreExt.jl

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -3,6 +3,7 @@ module TensorKitChainRulesCoreExt
33
using TensorOperations
44
using VectorInterface
55
using TensorKit
6+
using TensorKit: foreachblock
67
using ChainRulesCore
78
using LinearAlgebra
89
using TupleTools
@@ -11,6 +12,10 @@ import TensorOperations as TO
1112
using TensorOperations: promote_contract, tensoralloc_add, tensoralloc_contract
1213
using VectorInterface: promote_scale, promote_add
1314

15+
using MatrixAlgebraKit
16+
using MatrixAlgebraKit: TruncationStrategy,
17+
svd_compact_pullback!, eig_full_pullback!, eigh_full_pullback!
18+
1419
include("utility.jl")
1520
include("constructors.jl")
1621
include("linalg.jl")

ext/TensorKitChainRulesCoreExt/factorizations.jl

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1,5 +1,3 @@
1-
using MatrixAlgebraKit: svd_compact_pullback!, eig_full_pullback!, eigh_full_pullback!
2-
31
# Factorizations rules
42
# --------------------
53
function ChainRulesCore.rrule(::typeof(TensorKit.tsvd!), t::AbstractTensorMap;

0 commit comments

Comments
 (0)