Skip to content

[moe] polish code#4952

Merged
oahzxl merged 17 commits intohpcaitech:feature/MoEfrom
oahzxl:bench1
Oct 20, 2023
Merged

[moe] polish code#4952
oahzxl merged 17 commits intohpcaitech:feature/MoEfrom
oahzxl:bench1

Conversation

@oahzxl
Copy link
Copy Markdown
Contributor

@oahzxl oahzxl commented Oct 20, 2023

  • polish code
  • add hybrid zero2 support

@oahzxl oahzxl merged commit 45ac733 into hpcaitech:feature/MoE Oct 20, 2023
oahzxl added a commit to oahzxl/ColossalAI that referenced this pull request Oct 26, 2023
* doc

* update script

* update experts

* update optim in fsdp

* update kernel in sparse

* empty cache

* update script

* update bench

* update script

* remove epzero2

* fix

* update print

* update test script

* update script

* update manager

* update host

* update script
@oahzxl oahzxl deleted the bench1 branch November 6, 2023 02:21
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