Use permuteddims

Author: mtfishman

Project.toml

@@ -27,7 +27,7 @@ KroneckerArraysTensorProductsExt = "TensorProducts"
 Adapt = "4.3"
 BlockArrays = "1.6"
 BlockSparseArrays = "0.9"
-DerivableInterfaces = "0.5"
+DerivableInterfaces = "0.5.3"
 DiagonalArrays = "0.3.11"
 FillArrays = "1.13"
 GPUArraysCore = "0.2"

src/fillarrays/kroneckerarray.jl

@@ -96,8 +96,8 @@ function _copyto!!(dest::Delta{<:Any,N}, src::Delta{<:Any,N}) where {N}
   return dest
 end
 
-# TODO: Define `DerivableInterfaces.permuteddims` and overload that instead.
-function Base.PermutedDimsArray(a::Delta, perm)
+using DerivableInterfaces: DerivableInterfaces, permuteddims
+function DerivableInterfaces.permuteddims(a::Delta, perm)
   ax_perm = Base.PermutedDimsArrays.genperm(axes(a), perm)
   return Delta{eltype(a)}(ax_perm)
 end

src/kroneckerarray.jl

@@ -132,9 +132,9 @@ function _permutedims!!(dest::AbstractArray, src::AbstractArray, perm)
   return dest
 end
 
-# TODO: Define `DerivableInterfaces.permuteddims` and overload that instead.
-function Base.PermutedDimsArray(a::KroneckerArray, perm)
-  return PermutedDimsArray(arg1(a), perm) ⊗ PermutedDimsArray(arg2(a), perm)
+using DerivableInterfaces: DerivableInterfaces, permuteddims
+function DerivableInterfaces.permuteddims(a::KroneckerArray, perm)
+  return permuteddims(arg1(a), perm) ⊗ permuteddims(arg2(a), perm)
 end
 
 function Base.permutedims!(dest::KroneckerArray, src::KroneckerArray, perm)