[Prim]fix attrs loss in creating op

python/paddle/fluid/framework.py

@@ -2883,6 +2883,8 @@ def __init__(
             self._type = type
             self.attrs = attrs if attrs else {}
         else:
+            self.legacy_attrs = attrs if attrs else {}
+
             self.block = block
             self.desc = desc
             # note: not add self.attrs here:
@@ -3080,13 +3082,22 @@ def find_name(var_list, name):
                     )
 
             self.desc.check_attrs()
+
+            # record all attrs needed by creating op
+            for item in self.desc.attr_names():
+                self.legacy_attrs[item] = self.desc.attr(item)
+
             if self._has_kernel(type):
                 self.desc.infer_var_type(self.block.desc)
                 self.desc.infer_shape(self.block.desc)
 
     def _has_kernel(self, op_type):
         return op_type not in self.OP_WITHOUT_KERNEL_SET
 
+    def _get_runtime_attrs(self):
+        # record all attrs needed by creating op
+        return self.legacy_attrs
+
     def to_string(self, throw_on_error):
         """
         Get debug string.

python/paddle/incubate/autograd/primx.py

@@ -656,8 +656,10 @@ def expand_nested_list(xs):
                     outputs[op.output_names[i]] = op.output(op.output_names[i])
 
                 attrs = {}
-                for name in sorted(op.attr_names):
-                    attrs[name] = op.attr(name)
+                runtime_attrs = op._get_runtime_attrs()
+                for name in runtime_attrs.keys():
+                    attrs[name] = runtime_attrs[name]
+
                 from paddle.fluid.dygraph.base import param_guard
 
                 new_op_desc = block.desc.append_op()