Optimized 2D-by-1D broadcasting in optimized op_mul (#4965)
Detect that we are doing an elementwise multiplication for a 2D tensor and a 1D tensor. Dispatch to a vectorized kernel for this case. Differential Revision: [D61560826](https://our.internmc.facebook.com/intern/diff/D61560826/) Pull Request resolved: https://github.com/pytorch/executorch/pull/4808 --------- Co-authored-by: Scott Wolchok <swolchok@fb.com>
H
Hansong committed
58efb8b67e074e7822b380d100cfc0e6ca5bf053
Parent: ba06861
Committed by GitHub <noreply@github.com>
on 8/29/2024, 3:46:36 AM