add a fast path for EmbeddingBag calling FBGEMM (#36679)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/36679
Test Plan:
Imported from OSS
Unit tests:
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_failures_cpu
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_offsets_cpu
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_new_offsets_cpu
python test/run_test.py -i test_nn -- TestNNDeviceTypeCPU.test_EmbeddingBag_per_sample_weights_and_no_offsets_cpu
python test/test_nn.py TestNN.test_embeddingbag_from_pretrained
python test/test_nn.py TestNN.test_embeddingbag_from_pretrained_options
Finally run: python test/test_nn.py
Reviewed By: supriyar
Differential Revision: D21058034
Pulled By: xing-liu
fbshipit-source-id: 8fef39078132f63c406976d6b76c51f9ce573f90