Skip to content

Remove xla_mark_sharding_dynamo_custom_op#8823

Merged
tengyifei merged 3 commits intomasterfrom
yifeit/cleanup-dynamo
Mar 13, 2025
Merged

Remove xla_mark_sharding_dynamo_custom_op#8823
tengyifei merged 3 commits intomasterfrom
yifeit/cleanup-dynamo

Conversation

@tengyifei
Copy link
Copy Markdown
Collaborator

@tengyifei tengyifei commented Mar 12, 2025

AFAIK this is never used anywhere. There's a torch.ops.xla.dynamo_mark_sharding Python function that seems to be the way to mark sharding under dynamo.

@tengyifei tengyifei marked this pull request as ready for review March 12, 2025 06:50
@tengyifei tengyifei requested a review from lsy323 March 12, 2025 06:50
@tengyifei tengyifei force-pushed the yifeit/cleanup-dynamo branch from 33bcc0c to 05fb503 Compare March 12, 2025 06:52
AFAIK this is never used anywhere.
@tengyifei tengyifei force-pushed the yifeit/cleanup-dynamo branch from 05fb503 to ec71efa Compare March 12, 2025 06:53
@tengyifei tengyifei changed the base branch from yifeit/aot-mark-sharding to master March 12, 2025 06:53
@tengyifei tengyifei enabled auto-merge (squash) March 13, 2025 04:02
Copy link
Copy Markdown
Collaborator

@lsy323 lsy323 left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM, thanks!

@tengyifei tengyifei merged commit 27d455a into master Mar 13, 2025
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