Fix test for embedding sharding#4456

kausv wants to merge 1 commit intopytorch:mainpytorch/FBGEMM:mainfrom

kausv:export-D77963539kausv/FBGEMM:export-D77963539Copy head branch name to clipboard