未验证 提交 2b4977f2 编写于 作者: P pangyoki 提交者: GitHub

fix final_state_linear (#42820)

上级 9827c8b5
...@@ -65,7 +65,7 @@ static PyObject *eager_api_final_state_linear(PyObject *self, PyObject *args, ...@@ -65,7 +65,7 @@ static PyObject *eager_api_final_state_linear(PyObject *self, PyObject *args,
if (bias.initialized()) { if (bias.initialized()) {
auto mm_out = auto mm_out =
matmul_final_state_dygraph_function(x, weight, false, false); matmul_final_state_dygraph_function(x, weight, false, false);
auto out = add_final_state_dygraph_function(bias, mm_out); auto out = add_final_state_dygraph_function(mm_out, bias);
PyEval_RestoreThread(tstate); PyEval_RestoreThread(tstate);
tstate = nullptr; tstate = nullptr;
return ToPyObject(out); return ToPyObject(out);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册