提交 7c94fa9f 编写于 作者: Z zhaojiaying01

fix compilation problems

上级 7d1c7bb5
......@@ -27,7 +27,7 @@ bool GruUnitKernel<CPU, float>::Init(GruUnitParam<CPU> *param) {
template <>
void GruUnitKernel<CPU, float>::Compute(const GruUnitParam<CPU> &param) {
GruUnitCompute(param);
GruUnitCompute<float>(param);
}
template class GruUnitKernel<CPU, float>;
......
......@@ -47,7 +47,7 @@ void GruUnitCompute(const GruUnitParam<CPU>& param) {
gru_value.state_weight =
const_cast<P*>(weight_data + 2 * frame_size * frame_size);
gru_value.output_value = hidden->data<P>();
gru_value.prev_out_value = hidden_prev->data<P>();
gru_value.prev_out_value = const_cast<P*>(hidden_prev->data<P>());
gru_value.gate_value = gate->data<P>();
gru_value.reset_output_value = reset_hidden_prev->data<P>();
auto active_node = math::GetActivationType(param.Activation());
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册