adapt to TensorProducts

Project.toml

@@ -1,30 +1,25 @@
 name = "TensorAlgebra"
 uuid = "68bd88dc-f39d-4e12-b2ca-f046b68fcc6a"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.2.3"
+version = "0.2.1"
 
 [deps]
 ArrayLayouts = "4c555306-a7a7-4459-81d9-ec55ddd5c99a"
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
 EllipsisNotation = "da5c29d0-fa7d-589e-88eb-ea29b0a81949"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 MatrixAlgebraKit = "6c742aac-3347-4629-af66-fc926824e5e4"
+TensorProducts = "decf83d6-1968-43f4-96dc-fdb3fe15fc6d"
 TupleTools = "9d95972d-f1c8-5527-a6e0-b4b365fa01f6"
 TypeParameterAccessors = "7e5a90cf-f82e-492e-a09b-e3e26432c138"
 
-[weakdeps]
-GradedUnitRanges = "e2de450a-8a67-46c7-b59c-01d5a3d041c5"
-
-[extensions]
-TensorAlgebraGradedUnitRangesExt = "GradedUnitRanges"
-
 [compat]
 ArrayLayouts = "1.10.4"
-BlockArrays = "1.2.0"
+BlockArrays = "1.5.0"
 EllipsisNotation = "1.8.0"
-GradedUnitRanges = "0.1.0"
 LinearAlgebra = "1.10"
 MatrixAlgebraKit = "0.1.1"
+TensorProducts = "0.1.0"
 TupleTools = "1.6.0"
 TypeParameterAccessors = "0.2.1, 0.3"
 julia = "1.10"

src/fusedims.jl

@@ -1,3 +1,4 @@
+using TensorProducts: ⊗
 using .BaseExtensions: _permutedims, _permutedims!
 
 abstract type FusionStyle end
@@ -21,13 +22,6 @@ function fusedims(::ReshapeFusion, a::AbstractArray, axes::AbstractUnitRange...)
   return reshape(a, axes)
 end
 
-⊗(a::AbstractUnitRange) = a
-function ⊗(a1::AbstractUnitRange, a2::AbstractUnitRange, as::AbstractUnitRange...)
-  return ⊗(a1, ⊗(a2, as...))
-end
-⊗(a1::AbstractUnitRange, a2::AbstractUnitRange) = Base.OneTo(length(a1) * length(a2))
-⊗() = Base.OneTo(1)
-
 # Overload this version for most arrays
 function fusedims(a::AbstractArray, ax::AbstractUnitRange, axes::AbstractUnitRange...)
   return fusedims(FusionStyle(a), a, ax, axes...)

test/test_blockarrays_contract.jl

@@ -42,7 +42,7 @@ const elts = (Float32, Float64, Complex{Float32}, Complex{Float64})
     @test a_dest isa BlockedArray
     @test a_dest ≈ a_dest_dense
 
-    #  vector matrix
+    # vector matrix
     a_dest, dimnames_dest = contract(a3, (1, 2), a1, (2, -1, -2, 1))
     a_dest_dense, dimnames_dest_dense = contract(a3_dense, (1, 2), a1_dense, (2, -1, -2, 1))
     @test dimnames_dest == dimnames_dest_dense
@@ -88,7 +88,7 @@ const elts = (Float32, Float64, Complex{Float32}, Complex{Float64})
     @test a_dest isa BlockArray
     @test a_dest ≈ a_dest_dense
 
-    #  vector matrix
+    # vector matrix
     a_dest, dimnames_dest = contract(a3, (1, 2), a1, (2, -1, -2, 1))
     a_dest_dense, dimnames_dest_dense = contract(a3_dense, (1, 2), a1_dense, (2, -1, -2, 1))
     @test dimnames_dest == dimnames_dest_dense