xla
Expose mark_sharding_with_gradients as a public API.
#8826

Merged

Expose mark_sharding_with_gradients as a public API. #8826

tengyifei merged 8 commits into pytorch:master from iwknow:master

Expose mark_sharding_with_gradients as a public API.

bd23dbdd

Format modified files.

e74101bb

Fix typo in test.

99dc0dc0

Merge branch 'pytorch:master' into master

d03f414e

bhavya01 commented on 2025-03-14

tengyifei requested changes on 2025-03-14

Merge branch 'pytorch:master' into master

08f16275

Update the mark_sharding_with_gradients test to explicitly check the …

a9b7d08d

remove unused imports.

22c91788

fix typos and temporary settings.

75ef25ac

iwknow requested a review from

tengyifei 1 year ago

iwknow requested a review from

bhavya01 1 year ago

tengyifei approved these changes on 2025-03-17

tengyifei merged 5caaaee8 into master 1 year ago

Reviewers

tengyifei

bhavya01

Assignees

No one assigned

Labels

None yet

Milestone

No milestone