torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

embeddingbagcollection 16p vs 8p

Open imhandsome opened this issue 7 months ago • 0 comments

We found that the throughput of embeddingbagcollection 16p is only 1.2 times that of 8p. Are there any optimization measures?

imhandsome avatar Apr 22 '25 12:04 imhandsome