Skip to content

Commit

Permalink
Automated submodule update: FBGEMM (#112312)
Browse files Browse the repository at this point in the history 
This is an automated pull request to update the first-party submodule for [pytorch/FBGEMM](https://github.com/pytorch/FBGEMM).

New submodule commit: pytorch/FBGEMM@049f2a9

Test Plan: Ensure that CI jobs succeed on GitHub before landing.
Pull Request resolved: #112312
Approved by: https://github.com/malfet
  • Loading branch information
@facebook-github-bot @pytorchmergebot
facebook-github-bot authored and pytorchmergebot committed Nov 17, 2023
1 parent 7c35874 commit 6849d75
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion third_party/fbgemm
Submodule fbgemm updated 59 files
+1 −1 .github/scripts/fbgemm_gpu_test.bash
+2 −2 .github/scripts/test_torchrec.bash
+2 −2 .github/workflows/fbgemm_gpu_ci.yml
+1 −1 .github/workflows/fbgemm_gpu_pip.yml
+17 −15 fbgemm_gpu/CMakeLists.txt
+82 −140 fbgemm_gpu/cmake/Hip.cmake
+8 −0 fbgemm_gpu/codegen/embedding_backward_code_generator.py
+15 −4 fbgemm_gpu/codegen/embedding_backward_split_host_template.cpp
+2 −2 fbgemm_gpu/codegen/embedding_forward_quantized_host_cpu.cpp
+199 −0 fbgemm_gpu/codegen/embedding_forward_split_meta_template.cpp
+1 −3 fbgemm_gpu/codegen/split_embedding_codegen_lookup_invoker.template
+1 −1 fbgemm_gpu/docs/Doxyfile.in
+1 −1 fbgemm_gpu/fbgemm_gpu/__init__.py
+0 −106 fbgemm_gpu/fbgemm_gpu/sparse_operators.py
+286 −0 fbgemm_gpu/fbgemm_gpu/sparse_ops.py
+57 −20 fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_inference.py
+0 −2 fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training.py
+0 −2 fbgemm_gpu/fbgemm_gpu/ssd_split_table_batched_embeddings_ops.py
+3 −4 fbgemm_gpu/include/fbgemm_gpu/cumem_utils.h
+7 −0 fbgemm_gpu/include/fbgemm_gpu/dispatch_macros.h
+4 −2 fbgemm_gpu/include/fbgemm_gpu/sparse_ops.h
+1 −0 fbgemm_gpu/setup.py
+0 −0 fbgemm_gpu/src/embedding_inplace_ops/embedding_inplace_update.cu
+2 −10 fbgemm_gpu/src/embedding_inplace_ops/embedding_inplace_update_cpu.cpp
+0 −3 fbgemm_gpu/src/embedding_inplace_ops/embedding_inplace_update_gpu.cpp
+5 −2 fbgemm_gpu/src/embedding_inplace_ops/embedding_inplace_update_test.cpp
+0 −0 fbgemm_gpu/src/input_combine_ops/input_combine.cu
+0 −0 fbgemm_gpu/src/input_combine_ops/input_combine_cpu.cpp
+0 −0 fbgemm_gpu/src/input_combine_ops/input_combine_gpu.cpp
+40 −20 fbgemm_gpu/src/jagged_tensor_ops/jagged_tensor_ops_cpu.cpp
+3 −6 fbgemm_gpu/src/layout_transform_ops/layout_transform_ops.cu
+0 −0 fbgemm_gpu/src/layout_transform_ops/layout_transform_ops_cpu.cpp
+2 −3 fbgemm_gpu/src/layout_transform_ops/layout_transform_ops_gpu.cpp
+26 −0 fbgemm_gpu/src/memory_utils/common.cuh
+22 −0 fbgemm_gpu/src/memory_utils/common.h
+22 −0 fbgemm_gpu/src/memory_utils/memory_utils.cpp
+3 −12 fbgemm_gpu/src/memory_utils/memory_utils.cu
+6 −10 fbgemm_gpu/src/memory_utils/memory_utils_ops.cpp
+25 −0 fbgemm_gpu/src/memory_utils/memory_utils_ops.cu
+20 −4 fbgemm_gpu/src/merge_pooled_embedding_ops/merge_pooled_embedding_ops_cpu.cpp
+2 −24 fbgemm_gpu/src/merge_pooled_embedding_ops/merge_pooled_embedding_ops_gpu.cpp
+0 −0 fbgemm_gpu/src/metric_ops/metric_ops.cu
+0 −0 fbgemm_gpu/src/metric_ops/metric_ops.h
+0 −0 fbgemm_gpu/src/metric_ops/metric_ops_host.cpp
+3 −1 fbgemm_gpu/src/permute_pooled_embedding_ops/permute_pooled_embedding_ops_cpu.cpp
+21 −11 fbgemm_gpu/src/quantize_ops/quantize_fused_8bit_rowwise.cu
+2 −1 fbgemm_gpu/src/quantize_ops/quantize_ops_cpu.cpp
+4 −1 fbgemm_gpu/src/sparse_ops/sparse_block_bucketize_features.cu
+93 −30 fbgemm_gpu/src/sparse_ops/sparse_ops_cpu.cpp
+253 −76 fbgemm_gpu/src/sparse_ops/sparse_ops_gpu.cpp
+2 −1 fbgemm_gpu/src/sparse_ops/sparse_reorder_batched_ad.cu
+41 −210 fbgemm_gpu/test/failures_dict.json
+5 −90 fbgemm_gpu/test/failures_dict_fast.json
+23 −3 fbgemm_gpu/test/jagged_tensor_ops_test.py
+1 −0 fbgemm_gpu/test/merge_pooled_embeddings_test.py
+95 −48 fbgemm_gpu/test/quantize_ops_test.py
+284 −8 fbgemm_gpu/test/sparse_ops_test.py
+5 −1 fbgemm_gpu/test/test_utils.py
+17 −6 include/fbgemm/Fbgemm.h

0 comments on commit 6849d75

Please sign in to comment.