Embedding: Remove dispatch in parallel region (#60597)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/60597
Ref #56794
Test Plan: Imported from OSS
Reviewed By: jbschlosser
Differential Revision: D29446191
Pulled By: ngimel
fbshipit-source-id: d6ff010104ae621d5e3d9c269ed2b48407e71d67
