custom op use spmd rule

PaddlePaddle · Jan 5, 2024 · 7b8c0bc · 7b8c0bc
1 parent 72585ed
commit 7b8c0bc
Showing 1 changed file with 1 addition and 0 deletions.
diff --git 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;