Fix the trait for SubArrays

SciML · Nov 24, 2023 · 8536ca4 · 8536ca4
1 parent 770f141
commit 8536ca4
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/MaybeInplace.jl b/src/MaybeInplace.jl
@@ -191,6 +191,7 @@ struct CanSetindex <: AbstractMaybeSetindex end
 
 setindex_trait(::Number) = CannotSetindex()
 setindex_trait(::Array) = CanSetindex()
+setindex_trait(A::SubArray) = setindex_trait(parent(A))
 # In recent versions of Julia, this function has a type stable return type even without
 # overloading for sutom array types
 setindex_trait(A) = ifelse(can_setindex(A), CanSetindex(), CannotSetindex())