adapt to permuteblockeddims

Author: ogauthe

Project.toml

@@ -40,7 +40,7 @@ MacroTools = "0.5.13"
 MapBroadcast = "0.1.5"
 SparseArraysBase = "0.5"
 SplitApplyCombine = "1.2.3"
-TensorAlgebra = "0.3"
+TensorAlgebra = "0.3.1"
 Test = "1.10"
 TypeParameterAccessors = "0.2.0, 0.3"
 julia = "1.10"

ext/BlockSparseArraysTensorAlgebraExt/BlockSparseArraysTensorAlgebraExt.jl

@@ -3,7 +3,7 @@ module BlockSparseArraysTensorAlgebraExt
 using BlockSparseArrays: AbstractBlockSparseArray, blockreshape
 using TensorAlgebra:
   TensorAlgebra,
-  AbstractBlockPermutation,
+  BlockedTrivialPermutation,
   BlockedTuple,
   FusionStyle,
   ReshapeFusion,
@@ -16,11 +16,10 @@ function TensorAlgebra.FusionStyle(::AbstractBlockSparseArray, ::ReshapeFusion)
 end
 
 function TensorAlgebra.matricize(
-  ::BlockReshapeFusion, a::AbstractArray, biperm::AbstractBlockPermutation{2}
+  ::BlockReshapeFusion, a::AbstractArray, biperm::BlockedTrivialPermutation{2}
 )
-  a_perm = permutedims(a, Tuple(biperm))
-  new_axes = fuseaxes(axes(a_perm), biperm)
-  return blockreshape(a_perm, new_axes)
+  new_axes = fuseaxes(axes(a), biperm)
+  return blockreshape(a, new_axes)
 end
 
 function TensorAlgebra.unmatricize(

test/Project.toml

@@ -31,7 +31,7 @@ Random = "1"
 SafeTestsets = "0.1"
 SparseArraysBase = "0.5"
 Suppressor = "0.2"
-TensorAlgebra = "0.3"
+TensorAlgebra = "0.3.1"
 Test = "1"
 TestExtras = "0.3"
 TypeParameterAccessors = "0.3"