diff --git a/paddle/fluid/framework/fleet/ascend_wrapper.h b/paddle/fluid/framework/fleet/ascend_wrapper.h index 912d1b1c040b9..cba98699fc5e4 100644 --- a/paddle/fluid/framework/fleet/ascend_wrapper.h +++ b/paddle/fluid/framework/fleet/ascend_wrapper.h @@ -40,7 +40,7 @@ namespace framework { typedef ge::Graph AscendGraphDesc; #ifdef PADDLE_WITH_ASCEND_STRING -using AscendString = AscendString; +using AscendString = ge::AscendString; #else using AscendString = std::string; #endif diff --git a/paddle/fluid/pybind/ascend_wrapper_py.cc b/paddle/fluid/pybind/ascend_wrapper_py.cc index 11c90b8f90d27..cfab436a6446a 100644 --- a/paddle/fluid/pybind/ascend_wrapper_py.cc +++ b/paddle/fluid/pybind/ascend_wrapper_py.cc @@ -269,7 +269,7 @@ void BindAscendGraph(py::module *m) { #ifdef PADDLE_WITH_ASCEND_STRING .def("register_call_back_func", static_cast( + const char *, const ge::session::pCallBackFunc &)>( &ge::Session::RegisterCallBackFunc)) #else .def("register_call_back_func",