projects
/
platform
/
upstream
/
caffe.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
5e6ffd7
)
fix: add non-MKL sqrt (should have been included in ab33988)
author
Jeff Donahue
<jeff.donahue@gmail.com>
Thu, 13 Apr 2017 21:15:16 +0000
(14:15 -0700)
committer
Jeff Donahue
<jeff.donahue@gmail.com>
Thu, 13 Apr 2017 21:16:41 +0000
(14:16 -0700)
include/caffe/util/mkl_alternate.hpp
patch
|
blob
|
history
diff --git
a/include/caffe/util/mkl_alternate.hpp
b/include/caffe/util/mkl_alternate.hpp
index
79b2c32
..
8c2294c
100644
(file)
--- 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]))