SIGN IN SIGN UP

Add vectorized scalar path for single-element Tensor passed to optimized mul (#4964)

We are currently doing slow broadcasting for this case. After this diff, we should get nice vectorization.

Differential Revision: D61560825

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

---------

Co-authored-by: Scott Wolchok <swolchok@fb.com>
H
Hansong committed
ba06861dcb22a51626dec509fba249c1320f4564
Parent: 89ebeb0
Committed by GitHub <noreply@github.com> on 8/29/2024, 3:30:42 AM