From a7a2a4b18caeaf0803ecbf77d08c3cd6d26cbd79 Mon Sep 17 00:00:00 2001
From: mtfishman <mfishman@flatironinstitute.org>
Date: Mon, 24 Nov 2025 17:30:18 -0500
Subject: [PATCH 1/3] Upgrade to TensorAlgebra v0.5

---
 Project.toml                                  |  4 +-
 .../KroneckerArraysTensorAlgebraExt.jl        | 52 ++++++++++++-------
 test/Project.toml                             |  5 +-
 3 files changed, 38 insertions(+), 23 deletions(-)

diff --git a/Project.toml b/Project.toml
index 24cb025..4ecad6f 100644
--- a/Project.toml
+++ b/Project.toml
@@ -1,6 +1,6 @@
 name = "KroneckerArrays"
 uuid = "05d0b138-81bc-4ff7-84be-08becefb1ccc"
-version = "0.3.2"
+version = "0.3.3"
 authors = ["ITensor developers <support@itensor.org> and contributors"]
 
 [deps]
@@ -36,7 +36,7 @@ GPUArraysCore = "0.2"
 LinearAlgebra = "1.10"
 MapBroadcast = "0.1.10"
 MatrixAlgebraKit = "0.6"
-TensorAlgebra = "0.3.10, 0.4"
+TensorAlgebra = "0.5"
 TensorProducts = "0.1.7"
 TypeParameterAccessors = "0.4.2"
 julia = "1.10"
diff --git a/ext/KroneckerArraysTensorAlgebraExt/KroneckerArraysTensorAlgebraExt.jl b/ext/KroneckerArraysTensorAlgebraExt/KroneckerArraysTensorAlgebraExt.jl
index f997c8f..28f1618 100644
--- a/ext/KroneckerArraysTensorAlgebraExt/KroneckerArraysTensorAlgebraExt.jl
+++ b/ext/KroneckerArraysTensorAlgebraExt/KroneckerArraysTensorAlgebraExt.jl
@@ -1,8 +1,8 @@
 module KroneckerArraysTensorAlgebraExt
 
 using KroneckerArrays: KroneckerArrays, AbstractKroneckerArray, ⊗, kroneckerfactors
-using TensorAlgebra: TensorAlgebra, AbstractBlockPermutation, BlockedTrivialPermutation, FusionStyle,
-    matricize, unmatricize
+using TensorAlgebra: TensorAlgebra, AbstractBlockPermutation, BlockedTrivialPermutation,
+    FusionStyle, matricize, unmatricize
 
 struct KroneckerFusion{A <: FusionStyle, B <: FusionStyle} <: FusionStyle
     a::A
@@ -11,32 +11,44 @@ end
 KroneckerArrays.kroneckerfactors(style::KroneckerFusion) = (style.a, style.b)
 KroneckerArrays.kroneckerfactortypes(::Type{KroneckerFusion{A, B}}) where {A, B} = (A, B)
 
-TensorAlgebra.FusionStyle(a::AbstractKroneckerArray) = KroneckerFusion(FusionStyle.(kroneckerfactors(a))...)
+function TensorAlgebra.FusionStyle(a::AbstractKroneckerArray)
+    return KroneckerFusion(FusionStyle.(kroneckerfactors(a))...)
+end
 function matricize_kronecker(
-        style::KroneckerFusion, a::AbstractArray, biperm::AbstractBlockPermutation{2}
+        style::FusionStyle, a::AbstractArray, length1::Val, length2::Val
     )
-    return matricize(kroneckerfactors(style, 1), kroneckerfactors(a, 1), biperm) ⊗
-        matricize(kroneckerfactors(style, 2), kroneckerfactors(a, 2), biperm)
+    m1 = matricize(kroneckerfactors(style, 1), kroneckerfactors(a, 1), length1, length2)
+    m2 = matricize(kroneckerfactors(style, 2), kroneckerfactors(a, 2), length1, length2)
+    return m1 ⊗ m2
 end
 function TensorAlgebra.matricize(
-        style::KroneckerFusion, a::AbstractArray, biperm::AbstractBlockPermutation{2}
+        style::KroneckerFusion, a::AbstractArray, length1::Val, length2::Val
     )
-    return matricize_kronecker(style, a, biperm)
+    return matricize_kronecker(style, a, length1, length2)
 end
-# Fix ambiguity error.
-# TODO: Investigate rewriting the logic in `TensorAlgebra.jl` to avoid this.
-using TensorAlgebra: BlockedTrivialPermutation, unmatricize
-function TensorAlgebra.matricize(
-        style::KroneckerFusion, a::AbstractArray, biperm::BlockedTrivialPermutation{2}
+function unmatricize_kronecker(
+        style::FusionStyle,
+        m::AbstractMatrix,
+        codomain_axes::Tuple{Vararg{AbstractUnitRange}},
+        domain_axes::Tuple{Vararg{AbstractUnitRange}},
     )
-    return matricize_kronecker(style, a, biperm)
+    style1, style2 = kroneckerfactors(style)
+    m1, m2 = kroneckerfactors(m)
+    codomain1 = kroneckerfactors.(codomain_axes, 1)
+    codomain2 = kroneckerfactors.(codomain_axes, 2)
+    domain1 = kroneckerfactors.(domain_axes, 1)
+    domain2 = kroneckerfactors.(domain_axes, 2)
+    a1 = unmatricize(style1, m1, codomain1, domain1)
+    a2 = unmatricize(style2, m2, codomain2, domain2)
+    return a1 ⊗ a2
 end
-function unmatricize_kronecker(style::KroneckerFusion, a::AbstractArray, ax)
-    return unmatricize(kroneckerfactors(style, 1), kroneckerfactors(a, 1), kroneckerfactors.(ax, 1)) ⊗
-        unmatricize(kroneckerfactors(style, 2), kroneckerfactors(a, 2), kroneckerfactors.(ax, 2))
-end
-function TensorAlgebra.unmatricize(style::KroneckerFusion, a::AbstractArray, ax)
-    return unmatricize_kronecker(style, a, ax)
+function TensorAlgebra.unmatricize(
+        style::KroneckerFusion,
+        m::AbstractMatrix,
+        codomain_axes::Tuple{Vararg{AbstractUnitRange}},
+        domain_axes::Tuple{Vararg{AbstractUnitRange}},
+    )
+    return unmatricize_kronecker(style, m, codomain_axes, domain_axes)
 end
 
 end
diff --git a/test/Project.toml b/test/Project.toml
index c344ab9..45b84ef 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -19,6 +19,9 @@ TensorProducts = "decf83d6-1968-43f4-96dc-fdb3fe15fc6d"
 Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
 
+[sources]
+KroneckerArrays = {path = "/Users/mfishman/.julia/dev/KroneckerArrays"}
+
 [compat]
 Adapt = "4"
 Aqua = "0.8"
@@ -35,7 +38,7 @@ MatrixAlgebraKit = "0.6"
 SafeTestsets = "0.1"
 StableRNGs = "1.0"
 Suppressor = "0.2"
-TensorAlgebra = "0.3.10, 0.4"
+TensorAlgebra = "0.5"
 TensorProducts = "0.1.7"
 Test = "1.10"
 TestExtras = "0.3"

From 78e8360300e887956b43acd74120b6cefb9fb62d Mon Sep 17 00:00:00 2001
From: mtfishman <mfishman@flatironinstitute.org>
Date: Mon, 24 Nov 2025 17:31:15 -0500
Subject: [PATCH 2/3] Update gitignore

---
 .gitignore | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

diff --git a/.gitignore b/.gitignore
index 7085ca8..d5d9e4e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -1,18 +1,14 @@
-*.jl.*.cov
-*.jl.cov
-*.jl.mem
+*.cov
+*.mem
 *.o
 *.swp
 .DS_Store
 .benchmarkci
 .tmp
 .vscode/
-Manifest.toml
+LocalPreferences.toml
+Manifest*.toml
 benchmark/*.json
 dev/
-docs/LocalPreferences.toml
-docs/Manifest.toml
 docs/build/
 docs/src/index.md
-examples/LocalPreferences.toml
-test/LocalPreferences.toml

From 3a74c2fd69546e2199f9694fed37b60507403856 Mon Sep 17 00:00:00 2001
From: mtfishman <mfishman@flatironinstitute.org>
Date: Mon, 24 Nov 2025 17:43:50 -0500
Subject: [PATCH 3/3] Fix test Project

---
 test/Project.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/test/Project.toml b/test/Project.toml
index 45b84ef..32cd42a 100644
--- a/test/Project.toml
+++ b/test/Project.toml
@@ -20,7 +20,7 @@ Test = "8dfed614-e22c-5e08-85e1-65c5234f0b40"
 TestExtras = "5ed8adda-3752-4e41-b88a-e8b09835ee3a"
 
 [sources]
-KroneckerArrays = {path = "/Users/mfishman/.julia/dev/KroneckerArrays"}
+KroneckerArrays = {path = ".."}
 
 [compat]
 Adapt = "4"