FillArraysExt

Author: mtfishman

Project.toml

@@ -5,14 +5,17 @@ version = "0.3.1"
 
 [weakdeps]
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
+FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
 
 [extensions]
 FunctionImplementationsBlockArraysExt = "BlockArrays"
+FunctionImplementationsFillArraysExt = "FillArrays"
 FunctionImplementationsLinearAlgebraExt = "LinearAlgebra"
 
 [compat]
 BlockArrays = "1.4"
+FillArrays = "1.15"
 LinearAlgebra = "1.10"
 julia = "1.10"
 

ext/FunctionImplementationsFillArraysExt/FunctionImplementationsFillArraysExt.jl

@@ -0,0 +1,12 @@
+module FunctionImplementationsFillArraysExt
+
+using FillArrays: RectDiagonal
+using FunctionImplementations: FunctionImplementations
+
+function FunctionImplementations.permuteddims(a::RectDiagonal, perm)
+  (ndims(a) == length(perm) && isperm(perm)) ||
+    throw(ArgumentError("no valid permutation of dimensions"))
+  return RectDiagonal(parent(a), ntuple(d -> axes(a)[perm[d]], ndims(a)))
+end
+
+end

test/Project.toml

@@ -2,6 +2,7 @@
 Adapt = "79e6a3ab-5dfb-504d-930d-738a2a938a0e"
 Aqua = "4c88cf16-eb10-579e-8560-4a9242c79595"
 BlockArrays = "8e7c35d0-a365-5155-bbbb-fb81a777f24e"
+FillArrays = "1a297f60-69ca-5386-bcde-b61e274b549b"
 FunctionImplementations = "7c7cc465-9c6a-495f-bdd1-f42428e86d0c"
 JLArrays = "27aeb0d3-9eb9-45fb-866b-73c2ecf80fcb"
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"
@@ -16,6 +17,7 @@ FunctionImplementations = {path = ".."}
 Adapt = "4"
 Aqua = "0.8"
 BlockArrays = "1.4"
+FillArrays = "1.15"
 FunctionImplementations = "0.3"
 JLArrays = "0.3"
 LinearAlgebra = "1.10"

test/test_permuteddims.jl

@@ -1,3 +1,4 @@
+import FillArrays as FA
 import FunctionImplementations as FI
 import LinearAlgebra as LA
 using Test: @test, @testset
@@ -10,9 +11,15 @@ using Test: @test, @testset
         @test size(b) == (3, 2)
         @test b == permutedims(a, (2, 1))
     end
-    @testset "Diagonal" begin
+    @testset "LinearAlgebra.Diagonal" begin
         a = LA.Diagonal(randn(3))
         b = FI.permuteddims(a, (2, 1))
         @test b ≡ a
     end
+
+    @testset "FillArrays.RectDiagonal" begin
+        a = FA.RectDiagonal(randn(3), (3, 4))
+        @test FI.permuteddims(a, (1, 2)) ≡ a
+        @test FI.permuteddims(a, (2, 1)) ≡ FA.RectDiagonal(parent(a), (4, 3))
+    end
 end