SIGN IN SIGN UP

Simplify conv2d weight prepacking (>2x pipeline-creation speedup) (#3368)

Summary:
Pull Request resolved: https://github.com/pytorch/executorch/pull/3368

SS-JIA has previously written two implementations of convolution weights prepacking for CPU (before and after [PyTorch PR #84973](https://github.com/pytorch/pytorch/pull/84973)). Originally, I translated the second implementation to GPU since it is more readable. Now, I translate the first implementation to GPU and switch to it since it requires less steps.

Reviewed By: SS-JIA

Differential Revision: D56617129

fbshipit-source-id: b84533a45e1daf31f7fdec6707a518f3de57ce4c
J
Jorge Pineda committed
6c06f2624c3c6a13a3293ffce4fa06c189ecc179
Parent: 92b5aea
Committed by Facebook GitHub Bot <facebook-github-bot@users.noreply.github.com> on 4/26/2024, 9:53:40 PM