未验证 提交 52e38a00 编写于 作者: H houj04 提交者: GitHub

zero_copy_tensor unittest: support XPU. (#34670)

上级 4caf60df
...@@ -18,4 +18,4 @@ cc_library(fuse_all_reduce_op_pass SRCS fuse_all_reduce_op_pass.cc DEPS graph gr ...@@ -18,4 +18,4 @@ cc_library(fuse_all_reduce_op_pass SRCS fuse_all_reduce_op_pass.cc DEPS graph gr
cc_library(all_reduce_deps_pass SRCS all_reduce_deps_pass.cc DEPS all_reduce_op_handle graph graph_helper pass) cc_library(all_reduce_deps_pass SRCS all_reduce_deps_pass.cc DEPS all_reduce_op_handle graph graph_helper pass)
cc_library(backward_optimizer_op_deps_pass SRCS backward_optimizer_op_deps_pass.cc DEPS graph graph_helper pass) cc_library(backward_optimizer_op_deps_pass SRCS backward_optimizer_op_deps_pass.cc DEPS graph graph_helper pass)
cc_library(add_reader_dependency_pass SRCS add_reader_dependency_pass.cc DEPS graph graph_helper pass) cc_library(add_reader_dependency_pass SRCS add_reader_dependency_pass.cc DEPS graph graph_helper pass)
cc_library(fix_op_run_order_pass SRCS fix_op_run_order_pass DEPS graph graph_helper multi_devices_helper pass op_handle_base eager_deletion_op_handle) cc_library(fix_op_run_order_pass SRCS fix_op_run_order_pass.cc DEPS graph graph_helper multi_devices_helper pass op_handle_base eager_deletion_op_handle)
...@@ -137,6 +137,10 @@ TEST(Tensor, FillRandomDataAndCheck) { ...@@ -137,6 +137,10 @@ TEST(Tensor, FillRandomDataAndCheck) {
ASSERT_TRUE(FillRandomDataAndCheck(PlaceType::kNPU)); ASSERT_TRUE(FillRandomDataAndCheck(PlaceType::kNPU));
ASSERT_TRUE(SetPlaceAndCheck(PlaceType::kNPU)); ASSERT_TRUE(SetPlaceAndCheck(PlaceType::kNPU));
#endif #endif
#ifdef PADDLE_WITH_XPU
ASSERT_TRUE(FillRandomDataAndCheck(PlaceType::kXPU));
ASSERT_TRUE(SetPlaceAndCheck(PlaceType::kXPU));
#endif
} }
} // namespace paddle_infer } // namespace paddle_infer
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册