Skip to content

Commit

Permalink
Merge branch 'master' into patch-1
Browse files Browse the repository at this point in the history
  • Loading branch information
tjruwase authored Nov 20, 2024
2 parents 39ac81d + 83e4364 commit d2a0f26
Show file tree
Hide file tree
Showing 2 changed files with 3 additions and 3 deletions.
2 changes: 1 addition & 1 deletion deepspeed/runtime/zero/config.py
Original file line number Diff line number Diff line change
Expand Up @@ -311,7 +311,7 @@ class DeepSpeedZeroConfig(DeepSpeedConfigModel):
for efficient all_2_all_reduce comm
"""

mics_shard_size: int = Field(-1, new_param="mics_shard_size")
mics_shard_size: int = Field(-1, json_schema_extra={"new_param": "mics_shard_size"})

mics_hierarchical_params_gather: bool = False

Expand Down
4 changes: 2 additions & 2 deletions setup.py
Original file line number Diff line number Diff line change
Expand Up @@ -202,8 +202,8 @@ def op_enabled(op_name):
print(f'Install Ops={install_ops}')

# Write out version/git info.
git_hash_cmd = shlex.split("bash -c git rev-parse --short HEAD")
git_branch_cmd = shlex.split("bash -c git rev-parse --abbrev-ref HEAD")
git_hash_cmd = shlex.split("bash -c \"git rev-parse --short HEAD\"")
git_branch_cmd = shlex.split("bash -c \"git rev-parse --abbrev-ref HEAD\"")
if command_exists('git') and not is_env_set('DS_BUILD_STRING'):
try:
result = subprocess.check_output(git_hash_cmd)
Expand Down

0 comments on commit d2a0f26

Please sign in to comment.