diff options
| author | Jeff Donahue <jeff.donahue@gmail.com> | 2017-04-13 14:15:16 -0700 |
|---|---|---|
| committer | Jeff Donahue <jeff.donahue@gmail.com> | 2017-04-13 14:16:41 -0700 |
| commit | 3d5bed06a9b6b8a5dfd3db8da33f2fa3bc9a1213 (patch) | |
| tree | 0cd13e01e43e97fe231bbbdbca38db4a6e18477b | |
| parent | 5e6ffd71414d45ef0e0ba1ac55c7fe319664c686 (diff) | |
| download | caffe-3d5bed06a9b6b8a5dfd3db8da33f2fa3bc9a1213.tar.gz caffe-3d5bed06a9b6b8a5dfd3db8da33f2fa3bc9a1213.tar.bz2 caffe-3d5bed06a9b6b8a5dfd3db8da33f2fa3bc9a1213.zip | |
fix: add non-MKL sqrt (should have been included in ab33988)
| -rw-r--r-- | include/caffe/util/mkl_alternate.hpp | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/include/caffe/util/mkl_alternate.hpp b/include/caffe/util/mkl_alternate.hpp index 79b2c32d..8c2294c7 100644 --- a/include/caffe/util/mkl_alternate.hpp +++ b/include/caffe/util/mkl_alternate.hpp @@ -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])) |