Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Cleanup required_torch_version code and references. #5370

Merged
merged 12 commits into from
Apr 10, 2024

Conversation

loadams
Copy link
Contributor

@loadams loadams commented Apr 5, 2024

  • Move required_torch_version check from deepspeed.runtime.utils to deepspeed.utils.torch (newly created).
  • Remove unused duplicate definition from tests/unit/util.py.
  • Update all references to this function.
  • Switch checks in deepspeed/runtime/pipe/p2p.py to use this function.
  • Switch checks in deepspeed/comm/torch.py to use this function.

@loadams loadams changed the title Remove unused check, was previously moved to deepspeed/runtime/utils.py Cleanup required_torch_version code and references. Apr 8, 2024
@loadams loadams requested a review from lekurile April 8, 2024 18:45
@loadams loadams marked this pull request as ready for review April 8, 2024 19:26
@loadams loadams added this pull request to the merge queue Apr 10, 2024
Merged via the queue into master with commit 6dcced1 Apr 10, 2024
15 checks passed
rraminen pushed a commit to ROCm/DeepSpeed that referenced this pull request May 9, 2024
- Move `required_torch_version` check from deepspeed.runtime.utils to
deepspeed.utils.torch (newly created).
- Remove unused duplicate definition from `tests/unit/util.py`.
- Update all references to this function.
- Switch checks in `deepspeed/runtime/pipe/p2p.py` to use this function.
- Switch checks in `deepspeed/comm/torch.py` to use this function.

---------

Co-authored-by: Lev Kurilenko <[email protected]>
umchand pushed a commit to umchand/DeepSpeed that referenced this pull request May 20, 2024
- Move `required_torch_version` check from deepspeed.runtime.utils to
deepspeed.utils.torch (newly created).
- Remove unused duplicate definition from `tests/unit/util.py`.
- Update all references to this function.
- Switch checks in `deepspeed/runtime/pipe/p2p.py` to use this function.
- Switch checks in `deepspeed/comm/torch.py` to use this function.

---------

Co-authored-by: Lev Kurilenko <[email protected]>
dbyoung18 pushed a commit to dbyoung18/DeepSpeed that referenced this pull request Jun 11, 2024
- Move `required_torch_version` check from deepspeed.runtime.utils to
deepspeed.utils.torch (newly created).
- Remove unused duplicate definition from `tests/unit/util.py`.
- Update all references to this function.
- Switch checks in `deepspeed/runtime/pipe/p2p.py` to use this function.
- Switch checks in `deepspeed/comm/torch.py` to use this function.

---------

Co-authored-by: Lev Kurilenko <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants