Skip to content

Conversation

@vealocia
Copy link
Contributor

make partition_spec as tuple in case any scalar exists in model's parameters or intermediate outputs #8522

@tengyifei tengyifei self-requested a review January 25, 2025 00:17
@zpcore zpcore added the tpuci label Jan 25, 2025
@tengyifei tengyifei removed the tpuci label Jan 25, 2025
@tengyifei
Copy link
Collaborator

@zpcore i made all PR automatically run TPU checks so we don't need the tpuci label now

@tengyifei tengyifei merged commit 79e4e72 into pytorch:master Jan 25, 2025
12 checks passed
@tengyifei
Copy link
Collaborator

Thanks!

vealocia added a commit to vealocia/xla that referenced this pull request Jan 31, 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.

3 participants