diff --git a/paddle/fluid/framework/fleet/ascend_wrapper.h b/paddle/fluid/framework/fleet/ascend_wrapper.h index 912d1b1c040b97fb408685cec56898f8e4d09c62..cba98699fc5e40e758346d32a74406ad9a3c8bde 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 11c90b8f90d276f4183c04025a64fc73675db0a2..cfab436a6446a6c5258143be64dc6c6253d08335 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",