Make has_fast_linear_indexing based on IndexStyle and ndims

JuliaLang · Mar 24, 2023 · 3a9950a · 3a9950a
1 parent 8ca8ef3
commit 3a9950a
Showing 1 changed file with 2 additions and 5 deletions.
diff --git a/base/reducedim.jl b/base/reducedim.jl
@@ -226,11 +226,8 @@ end
 
 ## generic (map)reduction
 
-has_fast_linear_indexing(a::AbstractArrayOrBroadcasted) = false
-has_fast_linear_indexing(a::Array) = true
-has_fast_linear_indexing(::Union{Number,Ref,AbstractChar}) = true  # 0d objects, for Broadcasted
-has_fast_linear_indexing(bc::Broadcast.Broadcasted) =
-    all(has_fast_linear_indexing, bc.args)
+has_fast_linear_indexing(a::AbstractArrayOrBroadcasted) = IndexStyle(a) === IndexLinear()
+has_fast_linear_indexing(a::AbstractVector) = true
 
 function check_reducedims(R, A)
     # Check whether R has compatible dimensions w.r.t. A for reduction