custom op use spmd rule

PaddlePaddle · Jan 5, 2024 · 72585ed · 72585ed
1 parent ed6f32d
commit 72585ed
Show file tree

Hide file tree

Showing 17 changed files with 752 additions and 99 deletions.
diff --git a/cmake/inference_lib.cmake b/cmake/inference_lib.cmake
@@ -335,6 +335,12 @@ copy(
   DSTS ${PADDLE_INFERENCE_INSTALL_DIR}/paddle/include/paddle/phi/core/distributed/
 )
 
+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/paddle/extension.h b/paddle/extension.h
@@ -22,5 +22,4 @@ limitations under the License. */
 #endif
 // For initialization of DeviceContextPool and MemoryMethod
 #include "paddle/fluid/platform/init_phi.h"
-
 static paddle::InitPhi g_init_phi;