nccl-tests:NCCL测试

时间:2024-05-24 01:51:23
【文件属性】:

文件名称:nccl-tests:NCCL测试

文件大小:27KB

文件格式:ZIP

更新时间:2024-05-24 01:51:23

Cuda

NCCL测试 这些测试同时检查操作的性能和正确性。 建造 要构建测试,只需键入make 。 如果未在/ usr / local / cuda中安装CUDA,则可以指定CUDA_HOME。 同样,如果未在/ usr中安装NCCL,则可以指定NCCL_HOME。 $ make CUDA_HOME=/path/to/cuda NCCL_HOME=/path/to/nccl NCCL测试依赖MPI来处理多个进程,因此可以处理多个节点。 如果要编译具有MPI支持的测试,则需要设置MPI = 1并将MPI_HOME设置为MPI的安装路径。 $ make MPI=1 MPI_HOME=/path/to/mpi CUDA_HOME=/path/to/cuda NCCL_HOME=/path/to/nccl 用法 NCCL测试可以在多个进程,多个线程以及每个线程多个CUDA设备上运行。 进程数由MP


【文件预览】:
nccl-tests-master
----src()
--------reduce.cu(4KB)
--------common.h(8KB)
--------nccl1_compat.h(2KB)
--------common.cu(29KB)
--------all_gather.cu(4KB)
--------all_reduce.cu(4KB)
--------Makefile(2KB)
--------alltoall.cu(5KB)
--------reduce_scatter.cu(4KB)
--------broadcast.cu(4KB)
----doc()
--------PERFORMANCE.md(6KB)
----.gitignore(114B)
----Makefile(284B)
----README.md(3KB)
----LICENSE.txt(1KB)

网友评论