Skip to content

P
Paddle

BaiXuePrincess / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 1
Fork 0
P
Paddle
提交 02259575 编写于 作者: Q Qiao Longfei
浏览文件
操作

change elementwise_add to elementwise_add_to test=develop

上级 64406706
显示空白变更内容
内联 并排
Showing with 5 addition and 5 deletion
paddle/fluid/operators/math/selected_rows_functor.cc
浏览文件 @ 02259575
...@@ -233,7 +233,7 @@ template <typename DeviceContext, typename T> ...@@ -233,7 +233,7 @@ template <typename DeviceContext, typename T>
typename std::enable_if< typename std::enable_if<
std::is_floating_point<T>::value && std::is_floating_point<T>::value &&
std::is_same<DeviceContext, platform::CPUDeviceContext>::value>::type std::is_same<DeviceContext, platform::CPUDeviceContext>::value>::type
elementwise_add(const DeviceContext& ctx, BlasT<DeviceContext, T>* blas, elementwise_add_to(const DeviceContext& ctx, BlasT<DeviceContext, T>* blas,
size_t data_len, const T* in, T* out) { size_t data_len, const T* in, T* out) {
blas->AXPY(data_len, 1., in, out); blas->AXPY(data_len, 1., in, out);
} }
...@@ -242,7 +242,7 @@ template <typename DeviceContext, typename T> ...@@ -242,7 +242,7 @@ template <typename DeviceContext, typename T>
typename std::enable_if< typename std::enable_if<
!std::is_floating_point<T>::value && !std::is_floating_point<T>::value &&
std::is_same<DeviceContext, platform::CPUDeviceContext>::value>::type std::is_same<DeviceContext, platform::CPUDeviceContext>::value>::type
elementwise_add(const DeviceContext& ctx, BlasT<DeviceContext, T>* blas, elementwise_add_to(const DeviceContext& ctx, BlasT<DeviceContext, T>* blas,
size_t data_len, const T* in, T* out) { size_t data_len, const T* in, T* out) {
for (int64_t i = 0; i < data_len; i++) { for (int64_t i = 0; i < data_len; i++) {
out[i] += in[i]; out[i] += in[i];
...@@ -308,7 +308,7 @@ struct MergeAdd<platform::CPUDeviceContext, T> { ...@@ -308,7 +308,7 @@ struct MergeAdd<platform::CPUDeviceContext, T> {
for (size_t i = 0; i < input_rows.size(); i++) { for (size_t i = 0; i < input_rows.size(); i++) {
size_t out_i = rows_to_id[input_rows[i]]; size_t out_i = rows_to_id[input_rows[i]];
elementwise_add<platform::CPUDeviceContext, T>( elementwise_add_to<platform::CPUDeviceContext, T>(
context, &blas, static_cast<size_t>(input_width), context, &blas, static_cast<size_t>(input_width),
&input_data[i * input_width], &out_data[out_i * input_width]); &input_data[i * input_width], &out_data[out_i * input_width]);
} }
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或