From 7b8c0bc38ab94d326dec3bd7a4410a2344630be3 Mon Sep 17 00:00:00 2001 From: liuzhenhai93 Date: Fri, 5 Jan 2024 15:02:00 +0000 Subject: [PATCH] custom op use spmd rule --- paddle/extension.h | 1 + 1 file changed, 1 insertion(+) diff --git a/paddle/extension.h b/paddle/extension.h index c534a13a22d97..c60ee269fb173 100644 --- a/paddle/extension.h +++ b/paddle/extension.h @@ -22,4 +22,5 @@ limitations under the License. */ #endif // For initialization of DeviceContextPool and MemoryMethod #include "paddle/fluid/platform/init_phi.h" + static paddle::InitPhi g_init_phi;