-
Notifications
You must be signed in to change notification settings - Fork 24.1k
MPS: Eye op #78408
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
MPS: Eye op #78408
Conversation
🔗 Helpful links
❌ 5 New FailuresAs of commit b65e4a5 (more details on the Dr. CI page): Expand to see more
🕵️ 5 new failures recognized by patternsThe following CI failures do not appear to be due to upstream breakages
|
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
That sounds quite good.
We should mention in the PR description though that this is a temporary way to do testing until MPS gets integrated with the more generic testing we have.
@pytorchbot merge this please |
Merge failed due to Refusing to merge as mandatory check(s) Lint failed for rule superuser |
The build failures seem un-related to this checkin:
can we merge this @albanD |
@pytorchbot merge this please |
Hey @kulinseth. |
Summary: This can be used as a reference PR was to add Op in MPS backend. Pull Request resolved: #78408 Approved by: https://github.com/albanD Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/8552acbd7435eadb184e0cedc21df64d3bf30329 Reviewed By: seemethere Differential Revision: D36783514 Pulled By: seemethere fbshipit-source-id: 5022b45572f1729989eb1ce61f3ad7af5bd90adc
Summary: This can be used as a reference PR was to add Op in MPS backend. Pull Request resolved: pytorch#78408 Approved by: https://github.com/albanD Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/8552acbd7435eadb184e0cedc21df64d3bf30329 Reviewed By: seemethere Differential Revision: D36783514 Pulled By: seemethere fbshipit-source-id: 5022b45572f1729989eb1ce61f3ad7af5bd90adc
Summary: This can be used as a reference PR was to add Op in MPS backend. Pull Request resolved: pytorch#78408 Approved by: https://github.com/albanD Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/8552acbd7435eadb184e0cedc21df64d3bf30329 Reviewed By: seemethere Differential Revision: D36783514 Pulled By: seemethere fbshipit-source-id: 5022b45572f1729989eb1ce61f3ad7af5bd90adc
This can be used as a reference PR was to add Op in MPS backend. Pull Request resolved: pytorch#78408 Approved by: https://github.com/albanD
This can be used as a reference PR was to add Op in MPS backend. Pull Request resolved: #78408 Approved by: https://github.com/albanD (cherry picked from commit 8552acb)
MPSGraphTensor* outputTensor = [mpsGraph bandPartWithTensor:onesTensor | ||
numLower:0 | ||
numUpper:0 | ||
name:nil]; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
To check understanding —the mapping of the native operation here is calling the MPS bandPartWithTensor
function with a tensor full of 1.0
and the shape of result
to implement the eye
operation of producing a diagonal vector full of ones?
This adds an MPS backend implementation for `aten::angle` and `aten::angle_out` (mentioned in issue #77764), following the example #78408. Pull Request resolved: #143449 Approved by: https://github.com/malfet
This can be used as a reference PR was to add Op in MPS backend.