PaddlePaddle
diff --git a/‎paddle/fluid/distributed/auto_parallel/spmd_rules/layer_norm_spmd_rule.cc
Copy file name to clipboard
-280Lines changed: 0 additions & 280 deletions b/‎paddle/fluid/distributed/auto_parallel/spmd_rules/layer_norm_spmd_rule.cc
Copy file name to clipboard
-280Lines changed: 0 additions & 280 deletions
diff --git a/‎paddle/fluid/distributed/auto_parallel/spmd_rules/layer_norm_spmd_rule.h
Copy file name to clipboard
-41Lines changed: 0 additions & 41 deletions b/‎paddle/fluid/distributed/auto_parallel/spmd_rules/layer_norm_spmd_rule.h
Copy file name to clipboard
-41Lines changed: 0 additions & 41 deletions
diff --git a/‎paddle/fluid/distributed/auto_parallel/spmd_rules/rules.h
Copy file name to clipboard
-4Lines changed: 0 additions & 4 deletions b/‎paddle/fluid/distributed/auto_parallel/spmd_rules/rules.h
Copy file name to clipboard
-4Lines changed: 0 additions & 4 deletions
diff --git a/‎paddle/phi/core/distributed/auto_parallel/inferspmd_utils.cc
Copy file name to clipboard
+3Lines changed: 3 additions & 0 deletions b/‎paddle/phi/core/distributed/auto_parallel/inferspmd_utils.cc
Copy file name to clipboard
+3Lines changed: 3 additions & 0 deletions
diff --git a/‎paddle/phi/core/distributed/auto_parallel/inferspmd_utils.h
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions b/‎paddle/phi/core/distributed/auto_parallel/inferspmd_utils.h
Copy file name to clipboard
+2Lines changed: 2 additions & 0 deletions
@@ -17,7 +17,6 @@
 #include "paddle/fluid/distributed/auto_parallel/spmd_rules/common.h"
 #include "paddle/fluid/distributed/auto_parallel/spmd_rules/cross_entropy_with_softmax_spmd_rule.h"
 #include "paddle/fluid/distributed/auto_parallel/spmd_rules/embedding_spmd_rule.h"
-#include "paddle/fluid/distributed/auto_parallel/spmd_rules/layer_norm_spmd_rule.h"
 #include "paddle/fluid/distributed/auto_parallel/spmd_rules/replicated_spmd_rule.h"
 #include "paddle/fluid/distributed/auto_parallel/spmd_rules/reshape_spmd_rule.h"
 #include "paddle/fluid/distributed/auto_parallel/spmd_rules/softmax_spmd_rule.h"
@@ -29,9 +28,6 @@ namespace paddle {
 namespace distributed {
 namespace auto_parallel {
 
-// layer_norm rule
-REGISTER_SPMD_RULE(layer_norm, LayerNormSPMDRule);
-
 // replicated rule
 REGISTER_SPMD_RULE(replicated, ReplicatedSPMDRule);
 
 
@@ -53,6 +53,9 @@ AttrType InferSpmdContext::AttrAt(size_t idx) const {
   }
 }
 
+template float InferSpmdContext::AttrAt(size_t idx) const;
+template int InferSpmdContext::AttrAt(size_t idx) const;
+
 template <>
 bool InferSpmdContext::AttrAt(size_t idx) const {
   try {
 
@@ -153,6 +153,8 @@ struct InferSpmdFnImpl<Return (*)(Args...), infer_spmd_fn> {
 
   // TODO(chenweihang): support other attr type later as needed
   PD_SPECIALIZE_InferSpmdFnCallHelper_FOR_ATTRIBUTE(bool);
+  PD_SPECIALIZE_InferSpmdFnCallHelper_FOR_ATTRIBUTE(int);
+  PD_SPECIALIZE_InferSpmdFnCallHelper_FOR_ATTRIBUTE(float);
   PD_SPECIALIZE_InferSpmdFnCallHelper_FOR_CONST_ATTRIBUTE_REF(std::vector<int>);
   PD_SPECIALIZE_InferSpmdFnCallHelper_FOR_CONST_ATTRIBUTE_REF(
       std::vector<int64_t>);
Original file line number	Diff line number	Diff line change
`@@ -53,6 +53,9 @@ AttrType InferSpmdContext::AttrAt(size_t idx) const {`
`53`	`53`	`}`
`54`	`54`	`}`
`55`	`55`
	`56`	`+template float InferSpmdContext::AttrAt(size_t idx) const;`
	`57`	`+template int InferSpmdContext::AttrAt(size_t idx) const;`
	`58`	`+`
`56`	`59`	`template <>`
`57`	`60`	`bool InferSpmdContext::AttrAt(size_t idx) const {`
`58`	`61`	`try {`