fix _message of protobuf (#49673)

PaddlePaddle · Feb 21, 2023 · 7fe44fe · 7fe44fe
1 parent b39afb1
commit 7fe44fe
Showing 1 changed file with 7 additions and 1 deletion.
diff --git a/python/paddle/distributed/fleet/base/distributed_strategy.py b/python/paddle/distributed/fleet/base/distributed_strategy.py
@@ -22,6 +22,12 @@
 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 @@ def __repr__(self):
                             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):