From 52e38a00f91b665df16efea347ad366fa1186349 Mon Sep 17 00:00:00 2001 From: houj04 <35131887+houj04@users.noreply.github.com> Date: Fri, 6 Aug 2021 19:52:53 +0800 Subject: [PATCH] zero_copy_tensor unittest: support XPU. (#34670) --- .../framework/ir/multi_devices_graph_pass/CMakeLists.txt | 2 +- paddle/fluid/inference/api/details/zero_copy_tensor_test.cc | 4 ++++ 2 files changed, 5 insertions(+), 1 deletion(-) diff --git a/paddle/fluid/framework/ir/multi_devices_graph_pass/CMakeLists.txt b/paddle/fluid/framework/ir/multi_devices_graph_pass/CMakeLists.txt index f945ddbd5d..6764799d82 100644 --- a/paddle/fluid/framework/ir/multi_devices_graph_pass/CMakeLists.txt +++ b/paddle/fluid/framework/ir/multi_devices_graph_pass/CMakeLists.txt @@ -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(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(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) diff --git a/paddle/fluid/inference/api/details/zero_copy_tensor_test.cc b/paddle/fluid/inference/api/details/zero_copy_tensor_test.cc index f5140547b8..0c092a8684 100644 --- a/paddle/fluid/inference/api/details/zero_copy_tensor_test.cc +++ b/paddle/fluid/inference/api/details/zero_copy_tensor_test.cc @@ -137,6 +137,10 @@ TEST(Tensor, FillRandomDataAndCheck) { ASSERT_TRUE(FillRandomDataAndCheck(PlaceType::kNPU)); ASSERT_TRUE(SetPlaceAndCheck(PlaceType::kNPU)); #endif +#ifdef PADDLE_WITH_XPU + ASSERT_TRUE(FillRandomDataAndCheck(PlaceType::kXPU)); + ASSERT_TRUE(SetPlaceAndCheck(PlaceType::kXPU)); +#endif } } // namespace paddle_infer -- GitLab