Skip to content

Commit

Permalink
Move FBGEMM to e5d0c9448774e6bc577e7f210ecbec56b7a69f10
Browse files Browse the repository at this point in the history
  • Loading branch information
@htyu
htyu committed Jun 17, 2024
1 parent bbeba0a commit b27dd7a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion submodules/FBGEMM
Submodule FBGEMM updated 28 files
+31 −36 .github/scripts/fbgemm_gpu_build.bash
+20 −16 .github/scripts/fbgemm_gpu_install.bash
+14 −3 .github/scripts/fbgemm_gpu_test.bash
+2 −2 .github/scripts/nova_dir.bash
+34 −18 .github/scripts/utils_pip.bash
+1 −1 .github/workflows/fbgemm_gpu_ci_cpu.yml
+1 −1 .github/workflows/fbgemm_gpu_ci_cuda.yml
+1 −1 .github/workflows/fbgemm_gpu_ci_rocm.yml
+3 −3 .github/workflows/fbgemm_gpu_pip.yml
+1 −1 .github/workflows/fbgemm_gpu_release_cpu.yml
+1 −1 .github/workflows/fbgemm_gpu_release_cuda.yml
+2 −0 fbgemm_gpu/codegen/training/backward/embedding_backward_split_host_cpu_template.cpp
+2 −0 fbgemm_gpu/codegen/training/backward/embedding_backward_split_host_template.cpp
+38 −0 fbgemm_gpu/docs/src/fbgemm_gpu-development/BuildInstructions.rst
+147 −117 fbgemm_gpu/experimental/gemm/triton_gemm/fp8_gemm.py
+3 −2 fbgemm_gpu/experimental/gen_ai/bench/ck_fp8_bench.py
+4 −1 fbgemm_gpu/experimental/gen_ai/gen_ai/__init__.py
+2 −2 fbgemm_gpu/experimental/gen_ai/test/attention/gqa_test.py
+27 −4 fbgemm_gpu/fbgemm_gpu/quantize_comm.py
+10 −1 fbgemm_gpu/fbgemm_gpu/split_table_batched_embeddings_ops_training.py
+45 −12 fbgemm_gpu/fbgemm_gpu/ssd_split_table_batched_embeddings_ops.py
+2 −0 fbgemm_gpu/include/fbgemm_gpu/permute_pooled_embedding_function.h
+15 −5 fbgemm_gpu/setup.py
+8 −2 fbgemm_gpu/src/ps_split_embeddings_cache/ps_split_table_batched_embeddings.cpp
+4 −2 fbgemm_gpu/src/ps_split_embeddings_cache/ps_table_batched_embeddings.h
+3 −0 fbgemm_gpu/test/jagged/common.py
+6 −1 fbgemm_gpu/test/jagged/failures_dict.json
+10 −2 fbgemm_gpu/test/tbe/training/forward_test.py

0 comments on commit b27dd7a

Please sign in to comment.