Skip to content

[booster] add low level zero plugin#3594

Merged
FrankLeeeee merged 10 commits intohpcaitech:mainfrom
ver217:feature/booster-lowlevelzero
Apr 26, 2023
Merged

[booster] add low level zero plugin#3594
FrankLeeeee merged 10 commits intohpcaitech:mainfrom
ver217:feature/booster-lowlevelzero

Conversation

@ver217
Copy link
Copy Markdown
Contributor

@ver217 ver217 commented Apr 18, 2023

📌 Checklist before creating the PR

  • I have created an issue for this PR for traceability
  • The title follows the standard format: [doc/gemini/tensor/...]: A concise description
  • I have added relevant tags if possible for us to better distinguish different PRs

🚨 Issue number

Link this PR to your issue with words like fixed to automatically close the linked issue upon merge

e.g. fixed #1234, closed #1234, resolved #1234

📝 What does this PR do?

Summarize your work here.
if you have any plots/diagrams/screenshots/tables, please attach them here.

Add low level zero plugin and corresponding test

💥 Checklist before requesting a review

  • I have linked my PR to an issue (instruction)
  • My issue clearly describes the problem/feature/proposal, with diagrams/charts/table/code if possible
  • I have performed a self-review of my code
  • I have added thorough tests.
  • I have added docstrings for all the functions/methods I implemented

⭐️ Do you enjoy contributing to Colossal-AI?

  • 🌝 Yes, I do.
  • 🌚 No, I don't.

Tell us more if you don't enjoy contributing to Colossal-AI.

@ver217 ver217 added Run Build and Test API related to API changes labels Apr 18, 2023
@ver217 ver217 requested review from 1SAA and FrankLeeeee April 18, 2023 12:06
@github-actions
Copy link
Copy Markdown
Contributor

The code coverage for the changed files is 83%.

Click me to view the complete report
Name                                                           Stmts   Miss  Cover
----------------------------------------------------------------------------------
colossalai/booster/plugin/__init__.py                              5      0   100%
colossalai/booster/plugin/low_level_zero_plugin.py                92     14    85%
tests/test_booster/test_plugin/test_gemini_plugin.py              87     10    89%
tests/test_booster/test_plugin/test_low_level_zero_plugin.py      72     17    76%
tests/test_booster/test_plugin/test_torch_ddp_plugin.py           68     14    79%
----------------------------------------------------------------------------------
TOTAL                                                            324     55    83%

@github-actions
Copy link
Copy Markdown
Contributor

The code coverage for the changed files is 89%.

Click me to view the complete report
Name                                                           Stmts   Miss  Cover
----------------------------------------------------------------------------------
colossalai/booster/plugin/__init__.py                              5      0   100%
colossalai/booster/plugin/low_level_zero_plugin.py                92     14    85%
colossalai/zero/low_level/low_level_optim.py                     313     20    94%
tests/kit/model_zoo/diffusers/diffusers.py                        35      0   100%
tests/kit/model_zoo/torchaudio/torchaudio.py                      50      0   100%
tests/kit/model_zoo/torchvision/torchvision.py                    39      7    82%
tests/test_booster/test_plugin/test_gemini_plugin.py              87     10    89%
tests/test_booster/test_plugin/test_low_level_zero_plugin.py      72     17    76%
tests/test_booster/test_plugin/test_torch_ddp_plugin.py           68     14    79%
----------------------------------------------------------------------------------
TOTAL                                                            761     82    89%

@FrankLeeeee FrankLeeeee merged commit 4b3240c into hpcaitech:main Apr 26, 2023
@FrankLeeeee FrankLeeeee deleted the feature/booster-lowlevelzero branch April 26, 2023 06:37
@ver217 ver217 self-assigned this May 9, 2023
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

API related to API changes

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants