Skip to content

[MPS][BE] Combine two upsample_kernel_out_template into one#148211

Closed
malfet wants to merge 3 commits intogh/malfet/202/basefrom
gh/malfet/202/head
Closed

[MPS][BE] Combine two upsample_kernel_out_template into one#148211
malfet wants to merge 3 commits intogh/malfet/202/basefrom
gh/malfet/202/head

Conversation

@malfet
Copy link
Copy Markdown
Contributor

@malfet malfet commented Feb 28, 2025

Stack from ghstack (oldest at bottom):

  • First, by stopp inverting sizes and strides, i.e. passing them as is, but reading them in inverse order in the shader as 1st stride of 4D tensor is one used for batches, 2nd for channels and 3rd and 4th for spatial coordinates
  • Pass scales as float2 even in linear tensor
    Above allows one to collide two flavors upsample_kernel_out_template into one

First, by stopping inverting sizes and strides

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

pytorch-bot bot commented Feb 28, 2025

🔗 Helpful Links

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

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

⏳ No Failures, 12 Pending

As of commit 8217a71 with merge base 926b7b5 (image):
💚 Looks good so far! There are no failures yet. 💚

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

@pytorch-bot pytorch-bot bot added ciflow/mps Run MPS tests (subset of trunk) release notes: mps Release notes category labels Feb 28, 2025
@malfet malfet marked this pull request as draft February 28, 2025 20:33
…one"

First, by stopping inverting sizes and strides

[ghstack-poisoned]
…one"

First, by stopping inverting sizes and strides

[ghstack-poisoned]
malfet added a commit that referenced this pull request Feb 28, 2025
First, by stopping inverting sizes and strides

ghstack-source-id: d80a0a2
Pull Request resolved: #148211
@malfet malfet marked this pull request as ready for review February 28, 2025 22:58
@malfet malfet requested review from Skylion007 and dcci February 28, 2025 22:58
@malfet
Copy link
Copy Markdown
Contributor Author

malfet commented Mar 1, 2025

@pytorchbot merge -f "Another flaky failure"

@pytorchmergebot
Copy link
Copy Markdown
Collaborator

Merge started

Your change will be merged immediately since you used the force (-f) flag, bypassing any CI checks (ETA: 1-5 minutes). Please use -f as last resort and instead consider -i/--ignore-current to continue the merge ignoring current failures. This will allow currently pending tests to finish and report signal before the merge.

Learn more about merging in the wiki.

Questions? Feedback? Please reach out to the PyTorch DevX Team

Advanced Debugging
Check the merge workflow status
here

pytorchmergebot pushed a commit that referenced this pull request Mar 1, 2025
Not sure why tolerances were set like that, this logic was added in #104181 without much explanation
But if I'm to make a guess, it's likely due to the inaccuracy of bilinear op, that has since been replaced by shader
Pull Request resolved: #148224
Approved by: https://github.com/Skylion007, https://github.com/dcci
ghstack dependencies: #148154, #148187, #148211
@github-actions github-actions bot deleted the gh/malfet/202/head branch March 31, 2025 02:15
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

ciflow/mps Run MPS tests (subset of trunk) Merged release notes: mps Release notes category

Projects

None yet

Development

Successfully merging this pull request may close these issues.

3 participants