Skip to content

Move Permute layer to ops #6055

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.

Sign up for GitHub

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

Jump to bottom
Merged
merged 1 commit into from
May 20, 2022
Merged

Move Permute layer to ops #6055

merged 1 commit into from
May 20, 2022

Conversation

datumbox
Copy link
Contributor

Resolving #6053 (comment)

@datumbox datumbox mentioned this pull request May 19, 2022
Closed
24 tasks
NicolasHug
NicolasHug approved these changes May 20, 2022
View reviewed changes
Copy link
Member

@NicolasHug NicolasHug left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM.

For my own understanding, do we need a Permute module, or could we just rely on torch.permute? Looks like tihs Permute is a very simple wrapper

@datumbox
Copy link
Contributor Author

datumbox commented May 20, 2022 •
edited
Loading

@NicolasHug you are right that Permute is nothing but a wrapper for torch.permute. Unfortunately the latter can't be used as part of sequentials, which are very common in many of our models. I thought that instead of rewriting this over and over, it would be useful to have it as a shortcut on ops but I don't have strong opinions. Thoughts?

Another alternative is to move it to some _utils and make it private. That's also possible.

@datumbox
Copy link
Contributor Author

As discussed offline with @NicolasHug, it's OK to have this as an op.

@datumbox datumbox merged commit d57f929 into pytorch:main May 20, 2022
@datumbox datumbox deleted the ops/permute branch May 20, 2022 09:09
facebook-github-bot pushed a commit that referenced this pull request Jun 1, 2022
@YosuaMichael @facebook-github-bot
[fbsync] Move Permute layer to ops. (#6055)
7c7c118 
Reviewed By: NicolasHug

Differential Revision: D36760916

fbshipit-source-id: e71606bcdb48b74adf65c1def01c8f1ded22f8b7
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@NicolasHug NicolasHug NicolasHug approved these changes

Assignees

No one assigned

Labels

cla signed code quality module: ops

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@datumbox @NicolasHug @facebook-github-bot