From 88dfd067bf59d5ca4687672c1d17b9bb336f2b9d Mon Sep 17 00:00:00 2001 From: Leo Chen Date: Thu, 28 Jan 2021 15:46:54 +0800 Subject: [PATCH] Dev/fix ascend string (#30749) Dev/fix ascend string --- paddle/fluid/framework/fleet/ascend_wrapper.h | 2 +- paddle/fluid/pybind/ascend_wrapper_py.cc | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/paddle/fluid/framework/fleet/ascend_wrapper.h b/paddle/fluid/framework/fleet/ascend_wrapper.h index 912d1b1c04..cba98699fc 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 11c90b8f90..cfab436a64 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", -- GitLab