Merge branch 'master' into HEAD

Этот коммит содержится в:
Ke Wen
2019-06-25 13:39:08 -07:00
родитель 7c72dee660 37e4f8729e
Коммит 8e04d80382
2 изменённых файлов: 5 добавлений и 3 удалений
+4 -2
Просмотреть файл
@@ -880,10 +880,12 @@ static ncclResult_t getCpuGpuAffinity(int cudaDev, cpu_set_t* mask) {
path[PATH_MAX-1] = '\0';
int 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);
if (r > 0)
if (r > 0) {
affinityStr[r] = '\0';
NCCLCHECK(ncclStrToCpuset(affinityStr, mask));
}
close(fd);
free(cudaPath);
return ncclSuccess;