Added support for erf functions.

denizyuret · Oct 1, 2017 · 557ffbd · 557ffbd
1 parent b97c273
commit 557ffbd
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 7 deletions.
diff --git a/NEWS.md b/NEWS.md
@@ -4,6 +4,8 @@ Knet v0.8.5 Release Notes
 * Fixed memory leak with certain broadcast kernels.
 * Fixed dropout efficiency bug introduced in 0.8.4.
 * Fixed julia6 compat problem in examples/mnist.jl.
+* Added conditional support for SpecialFunctions.
+
 
 Knet v0.8.4 Release Notes
 =========================

diff --git a/src/unary.jl b/src/unary.jl
@@ -20,13 +20,6 @@ unary_ops = [
 "cospi",
 # "cyl_bessel_i0",
 # "cyl_bessel_i1",
-
-# "erf",     # Removed from base in julia6
-# "erfc",
-# "erfcinv",
-# "erfcx",
-# "erfinv",
-
 "exp",
 "exp10",
 "exp2",
@@ -69,6 +62,16 @@ unary_ops = [
 # "y1",
 ]
 
+if VERSION < v"0.6.0" || Pkg.installed("SpecialFunctions") != nothing
+    append!(unary_ops, [
+"erf",     # Removed from base in julia6
+"erfc",
+"erfcinv",
+"erfcx",
+"erfinv",
+                        ])
+end
+
 function unary_op(f, j=f, o...)
     J=broadcast_func(j)
     for S in (32,64)