pytorch

da860bda - Use correct WARP_SIZE for ROCm for EmbeddingBag

Use correct WARP_SIZE for ROCm for EmbeddingBag Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/24868 Differential Revision: D16936540 Pulled By: ezyang fbshipit-source-id: 08def671416888eb6ce57a690c7ff7743c21ad8c