Use DerivableInterfaces.permuteddims (#168)

Author: mtfishman

Project.toml

@@ -1,7 +1,7 @@
 name = "BlockSparseArrays"
 uuid = "2c9a651f-6452-4ace-a6ac-809f4280fbb4"
 authors = ["ITensor developers <[email protected]> and contributors"]
-version = "0.9.0"
+version = "0.9.1"
 
 [deps]
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
@@ -33,7 +33,7 @@ Adapt = "4.1.1"
 Aqua = "0.8.9"
 ArrayLayouts = "1.10.4"
 BlockArrays = "1.2.0"
-DerivableInterfaces = "0.5.2"
+DerivableInterfaces = "0.5.3"
 DiagonalArrays = "0.3"
 Dictionaries = "0.4.3"
 FillArrays = "1.13.0"

src/blocksparsearrayinterface/blocksparsearrayinterface.jl

@@ -22,6 +22,7 @@ using DerivableInterfaces:
   AbstractArrayInterface,
   DefaultArrayInterface,
   interface,
+  permuteddims,
   zero!
 using LinearAlgebra: Adjoint, Transpose
 using SparseArraysBase:
@@ -312,7 +313,7 @@ end
 # `blockisequal_map[!]`.
 # TODO: Maybe define a `BlockIsEqualInterface` for these kinds of functions.
 function blockisequal_permutedims!(a_dest::AbstractArray, a_src::AbstractArray, perm)
-  blocks(a_dest) .= blocks(PermutedDimsArray(a_src, perm))
+  blocks(a_dest) .= blocks(permuteddims(a_src, perm))
   return a_dest
 end
 
@@ -322,7 +323,7 @@ end
 @interface ::AbstractBlockSparseArrayInterface function Base.permutedims(
   a::AbstractArray, perm
 )
-  a_dest = similar(PermutedDimsArray(a, perm))
+  a_dest = similar(permuteddims(a, perm))
   # TODO: Maybe define this as `@interface BlockIsEqualInterface() permutedims!(...)`.
   blockisequal_permutedims!(a_dest, a, perm)
   return a_dest
@@ -334,7 +335,7 @@ end
 @interface ::AbstractBlockSparseArrayInterface function Base.permutedims!(
   a_dest::AbstractArray, a_src::AbstractArray, perm
 )
-  if all(blockisequal.(axes(a_dest), axes(PermutedDimsArray(a_src, perm))))
+  if all(blockisequal.(axes(a_dest), axes(permuteddims(a_src, perm))))
     # TODO: Maybe define this as `@interface BlockIsEqualInterface() permutedims!(...)`.
     blockisequal_permutedims!(a_dest, a_src, perm)
     return a_dest
@@ -403,15 +404,15 @@ end
 function SparseArraysBase.getstoredindex(
   a::SparsePermutedDimsArrayBlocks{<:Any,N}, index::Vararg{Int,N}
 ) where {N}
-  return PermutedDimsArray(
+  return permuteddims(
     getstoredindex(blocks(parent(a.array)), _getindices(index, _invperm(a.array))...),
     _perm(a.array),
   )
 end
 function SparseArraysBase.getunstoredindex(
   a::SparsePermutedDimsArrayBlocks{<:Any,N}, index::Vararg{Int,N}
 ) where {N}
-  return PermutedDimsArray(
+  return permuteddims(
     getunstoredindex(blocks(parent(a.array)), _getindices(index, _invperm(a.array))...),
     _perm(a.array),
   )