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