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

Distributed testing #6

Merged
merged 5 commits into from
Feb 4, 2020
Merged

Distributed testing #6

merged 5 commits into from
Feb 4, 2020

Conversation

ShadenSmith
Copy link
Contributor

Adds a @distributed_test decorator for easy distributed unit testing.

@ShadenSmith ShadenSmith mentioned this pull request Feb 4, 2020
@jeffra jeffra self-requested a review February 4, 2020 17:51
tests/unit/common.py Show resolved Hide resolved
@jeffra jeffra merged commit b61a221 into master Feb 4, 2020
@jeffra jeffra linked an issue Feb 4, 2020 that may be closed by this pull request
@jeffra jeffra deleted the distributed_test branch February 4, 2020 17:58
kouml pushed a commit to kouml/DeepSpeed that referenced this pull request Apr 3, 2020
* Adds distributed_test decorator and some unit tests.

* Setting NCCL backend.

* Parametrizes test.

* rank -> local_rank

* Temporarily disable CUDA initialization.
jithunnair-amd referenced this pull request in jithunnair-amd/DeepSpeed Apr 13, 2021
* Update skipIfRocm to add customizable reason string; update skipped unit tests

* Don't skip test_stage_to_global for now
rraminen pushed a commit to rraminen/DeepSpeed that referenced this pull request Apr 28, 2021
liamcli pushed a commit to determined-ai/DeepSpeed that referenced this pull request Sep 27, 2021
* Dist testing backend fixes, etc. (microsoft#708)

* set_batch_fn and remove old sanity check (microsoft#712)

* properly set engine.local_rank if it's set to -1

* Add executable permission to `ds_elastic` and `ds_report` in `bin`. (microsoft#711)

* Add executable permission to `ds_elastic` and `ds_report` in `bin`.

* Automatic `ds_elastic` formatting

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>

* local rank of -1 means not set (microsoft#720)

* bump to 0.3.11

* [launcher] look ma, no more zombies (microsoft#714)

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>

* Improve starred expressions (microsoft#696)

* Improve starred expressions

`deepspeed/profiling/flops_profiler/profiler.py` uses starred expressions
that are no longer valid with [PEP 617][1]. The new Python parser is in 3.9,
and this change allows DeepSpeed to run with the newest Python version. I have
not checked all locations that has this issue. However, this change allows me
to run simple examples.

[1]: https://www.python.org/dev/peps/pep-0617/

* Match style for "Improve starred expressions", although readability suffers

The style guide might need to be updated for this new use case of expressions.
Python [Issue 40631][1] includes more discussion on the change.

[1]: https://bugs.python.org/issue40631

Co-authored-by: Cheng Li <pistasable@gmail.com>

* Fixed typo in Readme. (microsoft#737)

* 1bit_adam dependencies (microsoft#742)

* Clickable screenshots (microsoft#746)

* Fix docstring

* Make screenshots clickable for easier viewing

* Add flops profiler tutorial (microsoft#682)

* work on flops profiler tutorial

* update flops profiler tutorial

* add flops profiler tutorial and fix names

* work on flops profiler tutorial

* update flops profiler tutorial

* add flops profiler tutorial and fix names

* fix tailing ws

* fix names

* remove multistep profiling and update docs

* fix cases where functionals and submodules coexist in a parent module, update readme

* fix typo

* always invoke post hook function

* fix module flops sum and update tests

* update tutorial

* Only initialize distributed if required (microsoft#734)

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>

Co-authored-by: Jeff Rasley <jerasley@microsoft.com>
Co-authored-by: Shaden Smith <Shaden.Smith@microsoft.com>
Co-authored-by: Jon Eyolfson <eyolfson@gmail.com>
Co-authored-by: Stas Bekman <stas00@users.noreply.github.com>
Co-authored-by: Cheng Li <pistasable@gmail.com>
Co-authored-by: TheDudeFromCI <thedudefromci@gmail.com>
Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
Co-authored-by: Sean Naren <sean@grid.ai>
delock referenced this pull request in delock/DeepSpeedSYCLSupport Sep 21, 2022
…t#60)

commit a8c119455d2ab3e58f9f6714f023fa0c0c0be817 (HEAD -> xpu-main, origin/xpu-main, origin/HEAD)
Author: Guo Yejun <yejun.guo@intel.com>
Date:   Thu Jul 21 01:18:39 2022 -0700

    add locate data

commit 79ee62539d76c3fefd5b1277b46077217eed71c7
Author: Liangliang-Ma <1906710196@qq.com>
Date:   Fri Aug 26 17:12:26 2022 +0800

    megatron with zero offload scripts (#6)

commit 38cb62ae4a8b468f27e3dddf929e3e53d302bb44
Author: Guo Yejun <yejun.guo@intel.com>
Date:   Fri Aug 19 08:32:34 2022 -0700

    ds_zero2_config_bf16.json&gpt-3.6b.sh: update bs per tile from 1 to 8

commit 65e6bb5e33d5a1244ce665b826f278244f5d0790
Author: Guo Yejun <yejun.guo@intel.com>
Date:   Tue Aug 2 20:49:38 2022 +0800

    ds_zero2_config_bf16.json: use the correct option flops_profiler (#4)
pengwa pushed a commit to pengwa/DeepSpeed that referenced this pull request Oct 14, 2022
sywangyi pushed a commit to sywangyi/DeepSpeed that referenced this pull request Mar 29, 2023
add sharded checkpoint loading for AutoTP path to reduce the peak mem…
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.

Distributed unit tests.
2 participants