polish

PaddlePaddle · Jan 2, 2024 · 1ca6d98 · 1ca6d98
1 parent 6aa5a27
commit 1ca6d98
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 1 deletion.
diff --git a/cmake/inference_lib.cmake b/cmake/inference_lib.cmake
@@ -328,10 +328,18 @@ copy(
   inference_lib_dist
   SRCS ${PADDLE_SOURCE_DIR}/paddle/phi/core/visit_type.h
   DSTS ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/paddle/phi/core/)
+
 copy(
   inference_lib_dist
   SRCS ${PADDLE_SOURCE_DIR}/paddle/phi/core/hostdevice.h
   DSTS ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/paddle/phi/core/)
+
+copy(
+  inference_lib_dist
+  SRCS ${PADDLE_SOURCE_DIR}/paddle/phi/core/distributed/auto_parallel/*.h
+  DSTS ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/paddle/phi/core/distributed/auto_parallel/
+)
+
 copy(
   inference_lib_dist
   SRCS ${PADDLE_SOURCE_DIR}/paddle/fluid/platform/init_phi.h

diff --git a/test/cpp/auto_parallel/custom_op_spmd_rule_test.cc b/test/cpp/auto_parallel/custom_op_spmd_rule_test.cc
@@ -63,7 +63,10 @@ TEST(CustomOp, Ctor) {
           infered_dist_attrs.first[0]));
   EXPECT_TRUE(paddle::holds_alternative<phi::distributed::TensorDistAttr>(
       infered_dist_attrs.second[0]));
-  auto& inputs_infer1 = paddle::get<1>(infered_dist_attrs.first[0]);
+  auto& inputs_infer1 =
+      PADDLE_GET_CONST(std::vector<phi::distributed::TensorDistAttr>,
+                       infered_dist_attrs.first[0]);
+
   for (auto e : inputs_infer1) {
     check_dim_mapping(e, {-1, 1, 0});
     check_partial_dims(e, {});