加入CODE CHINA

· 不限速    · 不限空间    · 不限人数    · 私仓免费

免费加入
README.md

MegRay

MegRay is a cross-platform communication library providing point-to-point and collective communication methods, such as send, recv, all_gather, all_reduce, reduce_scatter, reduce and broadcast. In the area of deep learning, these methods can be utilized for implementing distributed training framework, including data parallel and model parallel. Currently there are two backends, nccl and ucx, and only cuda platform is supported. In the future, algorithms on more platforms will be added.

Build

  1. prepare third party repositories.
./third_party/prepare.sh
  1. Make a directory for build.
mkdir build
cd build
  1. Generate build configurations by CMake.
cmake .. -DMEGRAY_TEST=ON
  1. Start to build
make

项目简介

A communication library for deep learning

发行版本

当前项目没有发行版本

贡献者 1

M Megvii Engine Team @Megvii Engine Team

开发语言

  • C++ 88.1 %
  • CMake 8.6 %
  • Cuda 2.8 %
  • C 0.4 %