-
Notifications
You must be signed in to change notification settings - Fork 26.3k
Added generation of transpose and dilated 2D and 3D for LongTensor #22594
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
facebook-github-bot
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.
@ifedan has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
aten/src/ATen/Dispatch.h
Outdated
| } \ | ||
| }() | ||
|
|
||
| #define AT_DISPATCH_FLOATING_TYPES_AND_LONG(TYPE, NAME, ...) \ |
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.
It might be nicer to add a AT_DISPATCH_FLOATING_TYPES_AND(SCALARTYPE, TYPE, NAME, ...
because there is already AT_DISPATCH_FLOATING_TYPES_AND_HALF and we might want a AT_DISPATCH_FLOATING_TYPES_AND_HALF_AND_LONG in the future...
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.
agreed.
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.
Fixed
gchanan
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 good if you fix up the dispatch macro, as suggested by @zou3519.
aten/src/ATen/Dispatch.h
Outdated
| } \ | ||
| }() | ||
|
|
||
| #define AT_DISPATCH_FLOATING_TYPES_AND_LONG(TYPE, NAME, ...) \ |
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.
agreed.
|
Also, please fix the title/commit message. This is clearly not implementing transpose. |
facebook-github-bot
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.
@ifedan has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.
…22594) Summary: Added implementations: transpose2D transpose3D dilated2D and dilated3D for LongTensor Pull Request resolved: pytorch/pytorch#22594 Differential Revision: D16155462 Pulled By: ifedan fbshipit-source-id: af57330314bc2c3e0a38b9e75105b20030a1f9bb
Added implementations: transpose2D transpose3D dilated2D and dilated3D for LongTensor