use axes to store the axes and not size

JuliaArrays · Jan 12, 2024 · 0e552c5 · 0e552c5
1 parent bc00c6a
commit 0e552c5
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 4 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,6 +1,6 @@
 name = "StaticArrays"
 uuid = "90137ffa-7385-5640-81b9-e52037218182"
-version = "1.9.0"
+version = "1.9.1"
 
 [deps]
 LinearAlgebra = "37e2e46d-f89d-539d-b4ee-838fcccc9c8e"

diff --git a/ext/StaticArraysChainRulesCoreExt.jl b/ext/StaticArraysChainRulesCoreExt.jl
@@ -15,12 +15,13 @@ end
 
 # Project SArray to SArray
 function ProjectTo(x::SArray{S, T}) where {S, T}
-    return ProjectTo{SArray}(; element = CRC._eltype_projectto(T), axes = Size(x))
+    return ProjectTo{SArray}(; element = CRC._eltype_projectto(T), axes = axes(x),
+        size = Size(x))
 end
 
 @inline _sarray_from_array(::Size{T}, dx::AbstractArray) where {T} = SArray{Tuple{T...}}(dx)
 
-(project::ProjectTo{SArray})(dx::AbstractArray) = _sarray_from_array(project.axes, dx)
+(project::ProjectTo{SArray})(dx::AbstractArray) = _sarray_from_array(project.size, dx)
 
 # Adjoint for SArray constructor
 function rrule(::Type{T}, x::Tuple) where {T <: SArray}

diff --git a/test/chainrules.jl b/test/chainrules.jl
@@ -1,4 +1,4 @@
-using StaticArrays, ChainRulesCore, ChainRulesTestUtils, JLArrays, Test
+using StaticArrays, ChainRulesCore, ChainRulesTestUtils, JLArrays, LinearAlgebra, Test
 
 @testset "ChainRules Integration" begin
     @testset "Projection" begin