未验证 提交 32ba8602 编写于 作者: W Wilber 提交者: GitHub

Enhance py_func error info message. (#26557)

上级 cb3f131f
...@@ -110,10 +110,12 @@ function(copy_part_of_thrid_party TARGET DST) ...@@ -110,10 +110,12 @@ function(copy_part_of_thrid_party TARGET DST)
SRCS ${GLOG_INCLUDE_DIR} ${GLOG_LIBRARIES} SRCS ${GLOG_INCLUDE_DIR} ${GLOG_LIBRARIES}
DSTS ${dst_dir} ${dst_dir}/lib) DSTS ${dst_dir} ${dst_dir}/lib)
if (WITH_CRYPTO)
set(dst_dir "${DST}/third_party/install/cryptopp") set(dst_dir "${DST}/third_party/install/cryptopp")
copy(${TARGET} copy(${TARGET}
SRCS ${CRYPTOPP_INCLUDE_DIR} ${CRYPTOPP_LIBRARIES} SRCS ${CRYPTOPP_INCLUDE_DIR} ${CRYPTOPP_LIBRARIES}
DSTS ${dst_dir} ${dst_dir}/lib) DSTS ${dst_dir} ${dst_dir}/lib)
endif()
set(dst_dir "${DST}/third_party/install/xxhash") set(dst_dir "${DST}/third_party/install/xxhash")
copy(${TARGET} copy(${TARGET}
......
...@@ -111,7 +111,8 @@ static void CallPythonFunc(py::object *callable, ...@@ -111,7 +111,8 @@ static void CallPythonFunc(py::object *callable,
out->set_lod(py_out_tensor->lod()); out->set_lod(py_out_tensor->lod());
out->ShareDataWith(*py_out_tensor); out->ShareDataWith(*py_out_tensor);
} catch (py::cast_error &) { } catch (py::cast_error &) {
PADDLE_THROW("The %d-th output must be LoDTensor", i); PADDLE_THROW(platform::errors::InvalidArgument(
"The %d-th output must be LoDTensor.", i));
} }
} }
} }
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册