fix _message of protobuf (#49673)

python/paddle/distributed/fleet/base/distributed_strategy.py

@@ -22,6 +22,12 @@ from paddle.distributed.fleet.utils.log_util import logger
 from paddle.fluid.framework import _global_flags
 from paddle.fluid.wrapped_decorator import wrap_decorator
 
+protobuf_version = google.protobuf.__version__
+if protobuf_version >= "4.21.0":
+    import google._upb._message as _message
+else:
+    import google.protobuf.pyext._message as _message
+
 __all__ = []
 
 non_auto_func_called = True
@@ -2497,7 +2503,7 @@ class DistributedStrategy:
                             for ff in config_fields:
                                 if isinstance(
                                     getattr(my_configs, ff.name),
-                                    google.protobuf.pyext._message.RepeatedScalarContainer,
+                                    _message.RepeatedScalarContainer,
                                 ):
                                     values = getattr(my_configs, ff.name)
                                     for i, v in enumerate(values):