From 535e45b265a0650cf6caa202501442647790c61c Mon Sep 17 00:00:00 2001 From: "Manuel R. Ciosici" Date: Thu, 30 Sep 2021 15:20:52 -0400 Subject: [PATCH] Fix typo in assert message --- deepspeed/runtime/zero/partition_parameters.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/runtime/zero/partition_parameters.py b/deepspeed/runtime/zero/partition_parameters.py index e9844d413735..50acacc7e70b 100755 --- a/deepspeed/runtime/zero/partition_parameters.py +++ b/deepspeed/runtime/zero/partition_parameters.py @@ -698,7 +698,7 @@ def _partition(self, param_list, force=False, has_been_updated=False): # self._param_status(param) def _partition_param(self, param, buffer=None, has_been_updated=False): - assert param.ds_status is not ZeroParamStatus.INFLIGHT, f" {param} Cannot parititon a param in flight" + assert param.ds_status is not ZeroParamStatus.INFLIGHT, f" {param} Cannot partition a param in flight" global reuse_buffers #print_rank_0(f"Param id {param.ds_id} status is {param.ds_status}")