yezhengmao/nccl
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge branch 'master' into HEAD

Browse Source
This commit is contained in:
Ke Wen 2019-06-25 13:39:08 -07:00
commit 8e04d80382
2 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@ -55,7 +55,7 @@ To install NCCL on the system, create a package then install it as root.
Debian/Ubuntu : Debian/Ubuntu :
```shell ```shell
$ # Install tools to create debian packages $ # Install tools to create debian packages
$ sudo apt install build-essential devscripts debhelper $ sudo apt install build-essential devscripts debhelper fakeroot
$ # Build NCCL deb package $ # Build NCCL deb package
$ make pkg.debian.build $ make pkg.debian.build
$ ls build/pkg/deb/ $ ls build/pkg/deb/

6
src/init.cc
View File

@ -880,10 +880,12 @@ static ncclResult_t getCpuGpuAffinity(int cudaDev, cpu_set_t* mask) {
path[PATH_MAX-1] = '\0'; path[PATH_MAX-1] = '\0';
int fd; int fd;
SYSCHECKVAL(open(path, O_RDONLY), "open", fd); SYSCHECKVAL(open(path, O_RDONLY), "open", fd);
char affinityStr[sizeof(cpu_set_t)*2]; char affinityStr[sizeof(cpu_set_t)*2 + 1];
int r = read(fd, affinityStr, sizeof(cpu_set_t)*2); int r = read(fd, affinityStr, sizeof(cpu_set_t)*2);
if (r > 0) if (r > 0) {
affinityStr[r] = '\0';
NCCLCHECK(ncclStrToCpuset(affinityStr, mask)); NCCLCHECK(ncclStrToCpuset(affinityStr, mask));
}
close(fd); close(fd);
free(cudaPath); free(cudaPath);
return ncclSuccess; return ncclSuccess;