From 5a46a4edd402d331a7211882dbf9359dc652b481 Mon Sep 17 00:00:00 2001 From: Gabriel Previato Date: Tue, 14 Feb 2023 00:05:35 -0300 Subject: [PATCH] Changing order of operations in mean function --- src/host/statistics.jl | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/host/statistics.jl b/src/host/statistics.jl index 3c6b0cc6..e40c1fa1 100644 --- a/src/host/statistics.jl +++ b/src/host/statistics.jl @@ -29,12 +29,12 @@ Statistics._mean(f, A::AbstractGPUArray, ::Colon) = sum(f, A) / length(A) function Statistics._mean(A::AbstractGPUArray, dims) T = float(eltype(A)) λ = convert(T, inv(_mean_denom(A, dims))) - sum(Base.Fix1(*,λ), A; dims) + sum(A; dims) .* λ end function Statistics._mean(f, A::AbstractGPUArray, dims) T = float(eltype(A)) λ = convert(T, inv(_mean_denom(A, dims))) - sum(Base.Fix1(*,λ) ∘ f, A; dims) + sum(f, A; dims) .* λ end function Statistics.covzm(x::AbstractGPUMatrix, vardim::Int=1; corrected::Bool=true)