bug fix

PaddlePaddle · Sep 26, 2023 · f1935cb · f1935cb
1 parent 3bf489b
commit f1935cb
Showing 1 changed file with 4 additions and 6 deletions.
diff --git a/paddle/phi/infermeta/spmd_rules/rules.h b/paddle/phi/infermeta/spmd_rules/rules.h
@@ -485,13 +485,11 @@ PD_REGISTER_SPMD_RULE(
     lookup_table_v2,
     PD_INFER_SPMD(phi::distributed::EmbeddingInferSpmd),
     PD_INFER_SPMD(phi::distributed::EmbeddingInferSpmdReverse));
-`
-    // split rule
-    PD_REGISTER_SPMD_RULE(
-        split,
-        PD_INFER_SPMD(phi::distributed::SplitInferSpmd),
-        PD_INFER_SPMD(phi::distributed::SplitInferSpmdReverse));
 
+// split rule
+PD_REGISTER_SPMD_RULE(split,
+                      PD_INFER_SPMD(phi::distributed::SplitInferSpmd),
+                      PD_INFER_SPMD(phi::distributed::SplitInferSpmdReverse));
 PD_REGISTER_SPMD_RULE(
     split_with_num,
     PD_INFER_SPMD(phi::distributed::SplitWithNumInferSpmd),