xla
Update MarkShardingFunction to be AOTAutograd traceable
#8821

Merged

Update MarkShardingFunction to be AOTAutograd traceable #8821

bhavya01 merged 1 commit into master from yifeit/aot-mark-sharding

tengyifei marked this pull request as ready for review 343 days ago

tengyifei requested a review from

bhavya01 343 days ago

tengyifei requested a review from

zpcore 343 days ago

tengyifei changed the base branch from hanq_dynamo2 to master 343 days ago

Update MarkShardingFunction to be AOTAutograd traceable

e4c19085

tengyifei force pushed from 25d35afc to e4c19085 343 days ago

bhavya01 approved these changes on 2025-03-12

bhavya01 merged c969276e into master 342 days ago

Reviewers

bhavya01

zpcore

Assignees

No one assigned

Labels

None yet

Milestone

No milestone