LICENSE.txt
MANIFEST.in
README.md
setup.cfg
setup.py
cgx_utils/__init__.py
cgx_utils/allreduce_hooks.py
pytorch_cgx.egg-info/PKG-INFO
pytorch_cgx.egg-info/SOURCES.txt
pytorch_cgx.egg-info/dependency_links.txt
pytorch_cgx.egg-info/top_level.txt
src/ProcessGroupCGX.cc
src/ProcessGroupCGX.h
src/mpi_allreduce_operations.cc
src/mpi_allreduce_operations.h
src/common/buffer.cc
src/common/buffer.h
src/common/common.h
src/common/communicator.h
src/common/compressor.cc
src/common/compressor.h
src/common/cuda_operations.cc
src/common/gpu_context.h
src/common/gpu_context_impl.cc
src/common/hip_operations.cc
src/common/layer.cc
src/common/layer.h
src/common/mpi_communicator.cc
src/common/mpi_communicator.h
src/common/mpi_context.cc
src/common/mpi_context.h
src/common/nccl_reduce.cc
src/common/nccl_reduce.h
src/common/reducer.cc
src/common/reducer.h
src/common/ring.cc
src/common/ring.h
src/common/scatter_reduce_allgather.cc
src/common/scatter_reduce_allgather.h
src/common/shm_communicator.cc
src/common/shm_communicator.h
src/common/shm_utils.cc
src/common/shm_utils.h
src/common/utils.cc
src/common/utils.h
src/common/compression/cuda_compression_operations.cu
src/common/compression/cuda_compression_operations.h
src/common/compression/gpu_common.h
src/common/compression/gpu_compression_operations.cc
src/common/compression/gpu_compression_operations.h
src/common/compression/gpu_def.h
src/common/compression/gpu_fp16_util.h
src/common/compression/gpu_rand.h
src/common/compression/hip_compression_operations.cc
src/common/compression/hip_compression_operations.h
test/test_cgx.py