fix: add non-MKL sqrt (should have been included in ab33988)
authorJeff Donahue <jeff.donahue@gmail.com>
Thu, 13 Apr 2017 21:15:16 +0000 (14:15 -0700)
committerJeff Donahue <jeff.donahue@gmail.com>
Thu, 13 Apr 2017 21:16:41 +0000 (14:16 -0700)
include/caffe/util/mkl_alternate.hpp

index 79b2c32..8c2294c 100644 (file)
@@ -37,6 +37,7 @@ extern "C" {
   }
 
 DEFINE_VSL_UNARY_FUNC(Sqr, y[i] = a[i] * a[i])
+DEFINE_VSL_UNARY_FUNC(Sqrt, y[i] = sqrt(a[i]))
 DEFINE_VSL_UNARY_FUNC(Exp, y[i] = exp(a[i]))
 DEFINE_VSL_UNARY_FUNC(Ln, y[i] = log(a[i]))
 DEFINE_VSL_UNARY_FUNC(Abs, y[i] = fabs(a[i]))