Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Revert "Image normalize operator - GPU support, 3D/4D inputs (#13802)" #14054

Conversation

sandeep-krishnamurthy
Copy link
Contributor

This reverts commit 3a1a80a.

Description

This PR is to revert #13802 where we added a normalize operator with batch support, gpu support and restructured to use kernel launch/map method.
I observed a performance drop in normalize operator on CPU:
Before the change:
Total time for 50000 (3,300,300) normalization - 67.39s
After the change:
Total time for 50000 images of shape (3,300,300) to do normalization - 104.09s

I am working on the fix. This PR is stop gap in case the performance fix PR is delayed.

@nswamy

@sandeep-krishnamurthy
Copy link
Contributor Author

PR with the fix - #14055
Do not merge this if the fix is merged.

@sandeep-krishnamurthy
Copy link
Contributor Author

Closing this PR in favor of the fix - #14055

Sign up for free to subscribe to this conversation on GitHub. Already have an account? Sign in.
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

1 participant