Fix typo (DistrubutedInputSpec -> DistributedInputSpec) (#61763)

co63oc · web-flow · commit 53d4978c55a2 · 2024-02-18T21:07:41.000+08:00
diff --git a/python/paddle/distributed/auto_parallel/static/dist_input_spec.py b/python/paddle/distributed/auto_parallel/static/dist_input_spec.py
@@ -20,7 +20,7 @@
 from .utils import convert_to_dims_mapping
 
 
-class DistrubutedInputSpec(InputSpec):
+class DistributedInputSpec(InputSpec):
     def __init__(
         self,
         shape,
@@ -38,13 +38,13 @@ def __init__(
     @classmethod
     def from_dtensor(cls, dtensor, name=None):
         """
-        Generates a DistrubutedInputSpec based on dist tensor.
+        Generates a DistributedInputSpec based on dist tensor.
 
         Args:
             dtensor: the dist tensor.
 
         Returns:
-            A DistrubutedInputSpec instance generated from dtensor.
+            A DistributedInputSpec instance generated from dtensor.
         """
         return cls(
             shape=dtensor.shape,
diff --git a/python/paddle/distributed/auto_parallel/static/engine.py b/python/paddle/distributed/auto_parallel/static/engine.py
@@ -45,7 +45,7 @@
 from .converter import Converter
 from .cost.estimate_cost import get_cost_from_engine
 from .dist_context import DistributedContext, get_default_distributed_context
-from .dist_input_spec import DistrubutedInputSpec
+from .dist_input_spec import DistributedInputSpec
 from .dist_loader import (
     DistributedDataLoader,
     DistributedDataLoaderFromGenerator,
@@ -294,14 +294,14 @@ def _prepare_data_spec_from_dataloader(self, dataloader):
                 assert item is not None, "Receive None input."
                 name = "input" + str(i)
                 inputs_spec.append(
-                    DistrubutedInputSpec.from_dtensor(item, name)
+                    DistributedInputSpec.from_dtensor(item, name)
                 )
         if labels is not None:
             for i, item in enumerate(labels):
                 assert item is not None, "Receive None input."
                 name = "label" + str(i)
                 labels_spec.append(
-                    DistrubutedInputSpec.from_dtensor(item, name)
+                    DistributedInputSpec.from_dtensor(item, name)
                 )
 
         inputs_spec = self._validate_spec(inputs_spec)
@@ -626,7 +626,7 @@ def _prepare_program(self, mode, init_parameters=True):
         self._has_prepared[mode] = True
 
     def _process_dist_input_specs(self):
-        if isinstance(self._inputs_spec[0], DistrubutedInputSpec):
+        if isinstance(self._inputs_spec[0], DistributedInputSpec):
 
             def _create_dist_input_var(input_var, input_spec):
                 dist_tensor = DistributedTensor(input_var)
@@ -1838,10 +1838,10 @@ def _validate_spec(self, specs):
         if specs is not None:
             for i, spec in enumerate(specs):
                 if not isinstance(spec, InputSpec) and not isinstance(
-                    spec, DistrubutedInputSpec
+                    spec, DistributedInputSpec
                 ):
                     raise TypeError(
-                        "'spec' must be object of class `paddle.static.InputSpec` or `DistrubutedInputSpec`."
+                        "'spec' must be object of class `paddle.static.InputSpec` or `DistributedInputSpec`."
                     )
                 if spec.name is None:
                     raise ValueError(
