Liang Luo c37095760d [torch distributed] Implementing all_gather_base (#56315) ... Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/56315 This diff implements the all_gather_base in pytorch distributed. Test Plan: dist.all_gather_base(output, input)... Reviewed By: agolynski, amylittleyang Differential Revision: D27488999 fbshipit-source-id: 937ec8bddf9527fa4d114f984d1d0f6a5b8c3936		2021-04-23 14:16:47 -07:00
..
no_python_abi_suffix_test
self_compiler_include_dirs_test
torch_test_cpp_extension
cpp_c10d_extension.cpp	[torch distributed] Implementing all_gather_base (#56315)	2021-04-23 14:16:47 -07:00
cpp_c10d_extension.hpp	[torch distributed] Implementing all_gather_base (#56315)	2021-04-23 14:16:47 -07:00
cpp_frontend_extension.cpp
cuda_extension.cpp
cuda_extension.cu
cuda_extension_kernel.cu
cuda_extension_kernel2.cu
cudnn_extension.cpp
doubler.h
extension.cpp
jit_extension.cpp
jit_extension2.cpp
msnpu_extension.cpp	Pass Scalar by reference (#53583)	2021-03-15 23:17:06 -07:00
rng_extension.cpp
setup.py	Fix ordered_dict.h for CUDA on Windows (#55275)	2021-04-07 23:43:35 -07:00
torch_library.cu