README.md 2.53 KB
Newer Older
Shaden Smith's avatar
Shaden Smith committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35

# DeepSpeed

## Installation


## Testing

DeepSpeed tracks two types of tests: unit tests and more costly model convergence tests.
The model convergence tests train
[DeepSpeedExamples](https://github.com/microsoft/DeepSpeedExamples/) and measure
end-to-end convergence and related metrics.  Unit tests are found in `tests/unit/` and
model convergence tests are found in `tests/model/`.

### Unit Tests
[PyTest](https://docs.pytest.org/en/latest/) is used to execute tests. PyTest can be
installed from PyPI via `pip install pytest`. Simply invoke `pytest` to run the unit
tests:

    pytest tests/unit/

You can also provide the `-v` flag to `pytest` to see additional information about the
tests.

### Model Tests
To execute model tests, first [install DeepSpeed](#installation). The
[DeepSpeedExamples](https://github.com/microsoft/DeepSpeedExamples/) repository is cloned
as part of this process. Next, execute the model test driver:

    cd tests/model/
    pytest run_sanity_check.py



## Contributing
36

yuxionghe's avatar
yuxionghe committed
37
38
DeepSpeed welcomes your contributions!

Shaden Smith's avatar
Shaden Smith committed
39
40
41
42
43
44
45
46
47
48
49
### Prerequisites

DeepSpeed uses [pre-commit](https://pre-commit.com/) to ensure that formatting is
consistent across DeepSpeed.  First, ensure that `pre-commit` is installed from either
installing DeepSpeed or `pip install pre-commit`.  Next, the pre-commit hooks must be
installed once before commits can be made:

    pre-commit install

Afterwards, our suite of formatting tests run automatically before each `git commit`. You
can also run these manually:
yuxionghe's avatar
yuxionghe committed
50

Shaden Smith's avatar
Shaden Smith committed
51
    pre-commit run --all-files
yuxionghe's avatar
yuxionghe committed
52
53


Shaden Smith's avatar
Shaden Smith committed
54
### Contributor License Agreement
yuxionghe's avatar
yuxionghe committed
55

Shaden Smith's avatar
Shaden Smith committed
56
57
58
59
This project welcomes contributions and suggestions.  Most contributions require you to
agree to a Contributor License Agreement (CLA) declaring that you have the right to, and
actually do, grant us the rights to use your contribution. For details, visit
https://cla.opensource.microsoft.com.
60

Shaden Smith's avatar
Shaden Smith committed
61
62
63
64
When you submit a pull request, a CLA bot will automatically determine whether you need
to provide a CLA and decorate the PR appropriately (e.g., status check, comment). Simply
follow the instructions provided by the bot. You will only need to do this once across
all repos using our CLA.
65

Shaden Smith's avatar
Shaden Smith committed
66
### Code of Conduct
yuxionghe's avatar
yuxionghe committed
67

Shaden Smith's avatar
Shaden Smith committed
68
69
70
71
72
This project has adopted the [Microsoft Open Source Code of
Conduct](https://opensource.microsoft.com/codeofconduct/).  For more information see the
[Code of Conduct FAQ](https://opensource.microsoft.com/codeofconduct/faq/) or contact
[opencode@microsoft.com](mailto:opencode@microsoft.com) with any additional questions or
comments.