Skip to content

Add mps support for maxpool3d#102148

Open
Berzeg wants to merge 2 commits intopytorch:mainfrom
Berzeg:mps-maxpool3d
Open

Add mps support for maxpool3d#102148
Berzeg wants to merge 2 commits intopytorch:mainfrom
Berzeg:mps-maxpool3d

Conversation

@Berzeg
Copy link
Copy Markdown

@Berzeg Berzeg commented May 24, 2023

Fixes #100674

Added mps support for forward / backward passes of maxpool3d.

As mentioned in the referenced issue, I'm utilizing the maxpool 4d op that is available from the metal library to achieve this.

cc: @mattiaspaul

@pytorch-bot pytorch-bot bot added ciflow/mps Run MPS tests (subset of trunk) release notes: mps Release notes category labels May 24, 2023
@pytorch-bot
Copy link
Copy Markdown

pytorch-bot bot commented May 24, 2023

🔗 Helpful Links

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

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

❗ 1 Active SEVs

There are 1 currently active SEVs. If your PR is affected, please view them below:

❌ 10 New Failures, 1 Unrelated Failure

As of commit 351390b with merge base 907cc6c (image):

NEW FAILURES - The following jobs have failed:

FLAKY - The following job failed but was likely due to flakiness present on trunk:

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

@soulitzer soulitzer added the triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module label May 25, 2023
@qqaatw
Copy link
Copy Markdown
Collaborator

qqaatw commented May 26, 2023

Hi, can you please add a test for it in test/test_mps.py?

Since there is an OpInfo for max_pool3d already, you can simply remove the following line and the consistency tests between cpu and mps will run automatically:

'nn.functional.max_pool3d': None,

@qqaatw
Copy link
Copy Markdown
Collaborator

qqaatw commented May 26, 2023

Also think you need to implement max_pool3d_with_indices* ops and adjust the decomp test in test_decomp.py since in MPS we treat w & w/o indices as separate kernels instead of sharing the same kernel and using CompositeImplicitAutograd. Let me know if you need any help!

@github-actions
Copy link
Copy Markdown
Contributor

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.

@github-actions github-actions bot added the Stale label Jul 25, 2023
@kulinseth
Copy link
Copy Markdown
Collaborator

Hi, can you please add a test for it in test/test_mps.py?

Since there is an OpInfo for max_pool3d already, you can simply remove the following line and the consistency tests between cpu and mps will run automatically:

'nn.functional.max_pool3d': None,

@Berzeg , thanks for the PR. Can you address comments from @qqaatw . Please let us know if you need help.

@qqaatw qqaatw removed the Stale label Aug 4, 2023
@github-actions
Copy link
Copy Markdown
Contributor

github-actions bot commented Oct 3, 2023

Looks like this PR hasn't been updated in a while so we're going to go ahead and mark this as Stale.
Feel free to remove the Stale label if you feel this was a mistake.
If you are unable to remove the Stale label please contact a maintainer in order to do so.
If you want the bot to never mark this PR stale again, add the no-stale label.
Stale pull requests will automatically be closed after 30 days of inactivity.

@github-actions github-actions bot added the Stale label Oct 3, 2023
@github-actions github-actions bot closed this Nov 2, 2023
@kulinseth kulinseth reopened this Jun 9, 2024
@kulinseth kulinseth requested a review from malfet as a code owner June 9, 2024 06:09
@kulinseth
Copy link
Copy Markdown
Collaborator

Re-opening this to take it forward.

@github-actions github-actions bot closed this Jul 10, 2024
@qqaatw qqaatw reopened this Jul 10, 2024
@qqaatw qqaatw added no-stale and removed Stale labels Jul 10, 2024
@rohanshad
Copy link
Copy Markdown

+1 for max_pool3d_with_indices and max_pool3d operations on mps

@Raman-RH
Copy link
Copy Markdown
Contributor

if this is pending, let me know. I'll be happy to contribute

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) no-stale open source release notes: mps Release notes category triaged This issue has been looked at a team member, and triaged and prioritized into an appropriate module

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Add support for MaxPool3D on the MPS backend

7 participants