lzhengning
/
JittorMirror
mirror of https://github.com/Jittor/Jittor
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

remove comment

This commit is contained in:
Dun Liang 2020-05-22 18:45:08 +08:00
parent 6e5876df65
commit e821e8c9ee
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
extern/cuda/nccl/src/nccl_warper.cc vendored
View File

@ -33,7 +33,7 @@ nccl_initer() {
if (mpi_local_rank >= device_count)
LOGf << "mpi_local_rank(">>mpi_local_rank>>") is larger than device_count("
>>device_count>>")";
nccl_device_id = mpi_local_rank; // % device_count;
nccl_device_id = mpi_local_rank;
LOGv << "NCCL init in device" << nccl_device_id << "local_rank" << mpi_local_rank;
checkCudaErrors(cudaSetDevice(nccl_device_id));
event_queue.run_sync([]() {