fix typo: change polciies to policies (#4090)

microsoft · Aug 4, 2023 · 4cde5da · 4cde5da
1 parent e831863
commit 4cde5da
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/deepspeed/module_inject/auto_tp.py b/deepspeed/module_inject/auto_tp.py
@@ -385,7 +385,7 @@ def update_mp_params(self, child):
                 setattr(child, param, param_val // self.mp_size)
         setattr(child, "replaced", True)
 
-    def update_linear_polciies(self):
+    def update_linear_policies(self):
         self.conv_linear_layer = False
         if self.linear_layer_setting is not None:
             self.linear_policies = {self.linear_layer_setting[0]: self._replace}

diff --git a/deepspeed/module_inject/replace_module.py b/deepspeed/module_inject/replace_module.py
@@ -273,7 +273,7 @@ def replace_wo_policy(module, all_reduce_linears, prefix="", state_dict=None):
         _autotp.set_tensor_parallel_config(config.tensor_parallel.tp_size, config.tensor_parallel.tp_group)
 
         # 3. Set linear policies
-        _autotp.update_linear_polciies()
+        _autotp.update_linear_policies()
 
         # 4. Replace modules
         return _autotp._replace_module(module)