Skip to content
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

[prototype] Remove _FT aliases from functional #6983

Merged
merged 3 commits into from
Nov 28, 2022

Conversation

datumbox
Copy link
Contributor

@datumbox datumbox commented Nov 28, 2022

Closes #6818

This PR eliminates the use of _FT aliases from prototype functional by porting over the last 3 remaining kernels. There might be some trivial speed improvements due to the elimination of unnecessary checks but nothing worth measuring IMO.

cc @vfdev-5 @bjuncek @pmeier

@datumbox datumbox requested review from vfdev-5 and pmeier November 28, 2022 12:10
@datumbox datumbox changed the title [prototype] Remove _FT usages from functional [prototype] Remove _FT aliases from functional Nov 28, 2022
Copy link
Collaborator

@pmeier pmeier left a comment

Choose a reason for hiding this comment

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

One minor nit otherwise LGTM!

torchvision/prototype/transforms/functional/_color.py Outdated Show resolved Hide resolved
Copy link
Collaborator

@vfdev-5 vfdev-5 left a comment

Choose a reason for hiding this comment

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

LGTM

@datumbox datumbox merged commit 4df1a85 into pytorch:main Nov 28, 2022
@datumbox datumbox deleted the prototype/remove_ft branch November 28, 2022 13:37
facebook-github-bot pushed a commit that referenced this pull request Dec 1, 2022
Summary:
* Remove `_FT` usages from functional

* Update error messages

Reviewed By: YosuaMichael

Differential Revision: D41648544

fbshipit-source-id: 096f5fa48cb1c320f8a23a25d2b048b268326746
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Projects
None yet
Development

Successfully merging this pull request may close these issues.

Performance improvements for transforms v2 vs. v1
4 participants