fix xlogx, xlogy

JuliaStats · May 19, 2020 · 9ec064b · 9ec064b
1 parent b7a603d
commit 9ec064b
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/basicfuns.jl b/src/basicfuns.jl
@@ -13,7 +13,7 @@ julia> StatsFuns.xlogx(0)
 """
 function xlogx(x)
     result = x * log(x)
-    ifelse(x > zero(x), result, zero(result))
+    ifelse(iszero(x), zero(result), result)
 end
 
 """
@@ -23,7 +23,7 @@ Return `x * log(y)` for `y > 0` with correct limit at `x = 0`.
 """
 function xlogy(x, y)
     result = x * log(y)
-    ifelse(x > zero(x), result, zero(result))
+    ifelse(iszero(x) && y ≥ 0, zero(result), result)
 end
 
 # The following bounds are precomputed versions of the following abstract