projects
/
platform
/
upstream
/
caffeonacl.git
/ commitdiff
commit
grep
author
committer
pickaxe
?
search:
re
summary
|
shortlog
|
log
|
commit
| commitdiff |
tree
raw
|
patch
| inline |
side by side
(parent:
348a338
)
Instantiate caffe_cpu_sign for float and double
author
Kai Li
<kaili_kloud@163.com>
Tue, 25 Feb 2014 11:21:18 +0000
(19:21 +0800)
committer
Kai Li
<kaili_kloud@163.com>
Sun, 23 Mar 2014 13:21:57 +0000
(21:21 +0800)
src/caffe/util/math_functions.cpp
patch
|
blob
|
history
diff --git
a/src/caffe/util/math_functions.cpp
b/src/caffe/util/math_functions.cpp
index
04a5228
..
bf24488
100644
(file)
--- a/
src/caffe/util/math_functions.cpp
+++ b/
src/caffe/util/math_functions.cpp
@@
-410,4
+410,10
@@
void caffe_gpu_asum<double>(const int n, const double* x, double* y) {
CUBLAS_CHECK(cublasDasum(Caffe::cublas_handle(), n, x, 1, y));
}
+template <>
+void caffe_cpu_sign<float>(const int n, const float* x, float* y);
+
+template <>
+void caffe_cpu_sign<double>(const int n, const double* x, double* y);
+
} // namespace caffe