Skip to content

[DTensor] Exclude Partial placements for boolean tensors in validator#174538

Closed
wconstab wants to merge 5 commits intogh/wconstab/525/basefrom
gh/wconstab/525/head
Closed

[DTensor] Exclude Partial placements for boolean tensors in validator#174538
wconstab wants to merge 5 commits intogh/wconstab/525/basefrom
gh/wconstab/525/head

Conversation

@wconstab
Copy link
Copy Markdown
Contributor

@wconstab wconstab commented Feb 8, 2026

Stack from ghstack (oldest at bottom):

Partial decomposition creates float local values (with offsets for
min/max, proportional splits for sum/avg) that lose boolean semantics
when cast back to bool by ops like where. This caused a massive
false positive explosion (134 "missing" rules for where) because
any Partial placement on a True boolean scalar produced non-zero
float values on all ranks, which all cast to True, making Partial
equivalent to Replicate for that sample but not in general.

Authored with Claude.

Partial decomposition creates float local values (with offsets for
min/max, proportional splits for sum/avg) that lose boolean semantics
when cast back to bool by ops like `where`. This caused a massive
false positive explosion (134 "missing" rules for `where`) because
any Partial placement on a True boolean scalar produced non-zero
float values on all ranks, which all cast to True, making Partial
equivalent to Replicate for that sample but not in general.

Authored with Claude.

[ghstack-poisoned]
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot Bot commented Feb 8, 2026

🔗 Helpful Links

🧪 See artifacts and rendered test results at hud.pytorch.org/pr/174538

Note: Links to docs will display an error until the docs builds have been completed.

⏳ No Failures, 39 Pending

As of commit 4dce57c with merge base f365425 (image):
💚 Looks good so far! There are no failures yet. 💚

This comment was automatically generated by Dr. CI and updates every 15 minutes.

…n validator"

Partial decomposition creates float local values (with offsets for
min/max, proportional splits for sum/avg) that lose boolean semantics
when cast back to bool by ops like `where`. This caused a massive
false positive explosion (134 "missing" rules for `where`) because
any Partial placement on a True boolean scalar produced non-zero
float values on all ranks, which all cast to True, making Partial
equivalent to Replicate for that sample but not in general.

Authored with Claude.

[ghstack-poisoned]
Copy link
Copy Markdown
Contributor

@pianpwk pianpwk left a comment

Choose a reason for hiding this comment

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

would this be solved by banning all True/False boolean tensors? it seems like a nice rule to discover

…n validator"

Partial decomposition creates float local values (with offsets for
min/max, proportional splits for sum/avg) that lose boolean semantics
when cast back to bool by ops like `where`. This caused a massive
false positive explosion (134 "missing" rules for `where`) because
any Partial placement on a True boolean scalar produced non-zero
float values on all ranks, which all cast to True, making Partial
equivalent to Replicate for that sample but not in general.

Authored with Claude.

[ghstack-poisoned]
…n validator"

Partial decomposition creates float local values (with offsets for
min/max, proportional splits for sum/avg) that lose boolean semantics
when cast back to bool by ops like `where`. This caused a massive
false positive explosion (134 "missing" rules for `where`) because
any Partial placement on a True boolean scalar produced non-zero
float values on all ranks, which all cast to True, making Partial
equivalent to Replicate for that sample but not in general.

Authored with Claude.

[ghstack-poisoned]
…n validator"

Partial decomposition creates float local values (with offsets for
min/max, proportional splits for sum/avg) that lose boolean semantics
when cast back to bool by ops like `where`. This caused a massive
false positive explosion (134 "missing" rules for `where`) because
any Partial placement on a True boolean scalar produced non-zero
float values on all ranks, which all cast to True, making Partial
equivalent to Replicate for that sample but not in general.

Authored with Claude.

[ghstack-poisoned]
@wconstab
Copy link
Copy Markdown
Contributor Author

squashed

@wconstab wconstab closed this Feb 11, 2026
sandy-gags pushed a commit to sandy-gags/pytorch that referenced this pull request Mar 12, 2026
Partial decomposition creates float local values (with offsets for
min/max, proportional splits for sum/avg) that lose boolean semantics
when cast back to bool by ops like `where`. This caused a massive
false positive explosion (134 "missing" rules for `where`) because
any Partial placement on a True boolean scalar produced non-zero
float values on all ranks, which all cast to True, making Partial
equivalent to Replicate for that sample but not in general.

Authored with Claude.

ghstack-source-id: 3708ac5
Pull Request resolved: pytorch/pytorch#174538
@github-actions github-actions Bot deleted the gh/wconstab/525/head branch March 14, 2026 02:24
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants