-
Notifications
You must be signed in to change notification settings - Fork 26.3k
[MPS] Workaround for uint8 gather bug #82315
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
Conversation
janeyx99
left a comment
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.
Looks correct
kulinseth
left a comment
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.
For now, its okay to have this workaround. But we need to investigate the generated Graph as to why uint8 data type is not support.
|
@pytorchbot merge |
|
@pytorchbot successfully started a merge and created land time checks. See merge status here and land check progress here |
Looks like MPSGraph generates a wrong shader for expanding column into a matrix, though datatype should not really matter for scatter-gather ops, should it? Also, fix minor typo: `c10::ScalarType:Char` is signed type, whereas `c10::ScalarType::Byte` is unsigned. Workarounds #82305 by aliasing tensors of `uint8` types to `int8` typed ones Pull Request resolved: #82315 Approved by: https://github.com/janeyx99, https://github.com/kulinseth
|
Hey @malfet. |
Summary: Looks like MPSGraph generates a wrong shader for expanding column into a matrix, though datatype should not really matter for scatter-gather ops, should it? Also, fix minor typo: `c10::ScalarType:Char` is signed type, whereas `c10::ScalarType::Byte` is unsigned. Workarounds #82305 by aliasing tensors of `uint8` types to `int8` typed ones Pull Request resolved: #82315 Approved by: https://github.com/janeyx99, https://github.com/kulinseth Test Plan: contbuild & OSS CI, see https://hud.pytorch.org/commit/pytorch/pytorch/47b03ad42e10b4bf6602c1d5bcb82079800e67b5 Reviewed By: osalpekar Differential Revision: D38234366 Pulled By: malfet fbshipit-source-id: 87b7652176c2e73c9e523995999e27e8e5a79381
And use convenient methods TODO was added by an accidental copy-n-paste of code from #82315 into #88532 Pull Request resolved: #118902 Approved by: https://github.com/kit1980
And use convenient methods TODO was added by an accidental copy-n-paste of code from #82315 into #88532 Pull Request resolved: #118902 Approved by: https://github.com/kit1980
Looks like MPSGraph generates a wrong shader for expanding column into a matrix, though datatype should not really matter for scatter-gather ops, should it?
Also, fix minor typo:
c10::ScalarType:Charis signed type, whereasc10::ScalarType::Byteis unsigned.Workarounds #82305 by aliasing tensors of
uint8types toint8typed ones