DeepSpeed
Add API to set a module as a leaf node when recursively setting Z3 hooks
#4966

Merged

Add API to set a module as a leaf node when recursively setting Z3 hooks #4966

tohtana merged 9 commits into master from tohtana/no_break_for_param_fetch

add api to stop breaking down modules for parameter fetching

5f025fcd

rename apis and add test

1caceec9

Merge branch 'master' into tohtana/no_break_for_param_fetch

e0acb89e

tohtana changed the title ~~Add API to stop breaking down modules for parameter fetching~~ Add API to set a module as a leaf node when recursively setting Z3 hooks 2 years ago

add default z3 leaf module class

e16ac33f

Merge branch 'tohtana/no_break_for_param_fetch' of github.com:microso…

205e5313

fix format

20908961

tohtana marked this pull request as ready for review 2 years ago

tohtana requested a review from

tjruwase 2 years ago

tohtana requested a review from

mrwyattii 2 years ago

tohtana requested a review from

awan-10 2 years ago

tohtana requested a review from

loadams 2 years ago

Merge branch 'master' into tohtana/no_break_for_param_fetch

3be82484

tjruwase commented on 2024-01-19

remove list of default leaf modules

e0fda1c8

Merge branch 'tohtana/no_break_for_param_fetch' of github.com:microso…

d89e951c

tjruwase approved these changes on 2024-01-19

tohtana merged 96c5a873 into master 2 years ago

tohtana deleted the tohtana/no_break_for_param_fetch branch 2 years ago

Reviewers

tjruwase

mrwyattii

awan-10

loadams

Assignees

No one assigned

Labels

None yet

Milestone

No milestone

DeepSpeed Add API to set a module as a leaf node when recursively setting Z3 hooks #4966 Merged

Add API to set a module as a leaf node when recursively setting Z3 hooks #4966

DeepSpeed
Add API to set a module as a leaf node when recursively setting Z3 hooks
#4966

Merged