PR #8723 Add autograd function for mark_sharding

Add autograd function for mark_sharding #8723

bhavya01 merged 5 commits into master from mark_sharding_autograd

Add autograd function for mark_sharding

061bb831

Add docs and fix linting

8f86189d

Fix lint with yapf==0.30.0

0a259dd1

Skip test if num devices is 1

f171e959

Remove empty space

17ee4d96

bhavya01 marked this pull request as ready for review 1 year ago

bhavya01 requested a review from

tengyifei 1 year ago

tengyifei commented on 2025-02-20

tengyifei approved these changes on 2025-02-20

bhavya01 merged 1acc987a into master 1 year ago

Reviewers

tengyifei

Assignees

No one assigned

Labels

None yet

Milestone

No milestone