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

Generate host-hash for P2P and SHM based on $(readlink /proc/self/ns/uts) + $(readlink /proc/self/ns/mnt) (#156)

Browse Source
This commit is contained in:
Alex Sergeev 2018-11-19 17:39:44 -08:00 committed by Sylvain Jeaugey
parent 3c6e25210b
commit d7a58cfa58
1 changed files with 12 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
src/misc/utils.cu
View File

@ -33,17 +33,24 @@ uint64_t getHash(const char* string) {
* that will be unique for both bare-metal and container instances
* Equivalent of a hash of;
*
* $(hostname) $(readlink /proc/self/ns/uts)
* $(hostname) $(readlink /proc/self/ns/uts) $(readlink /proc/self/ns/mnt)
*/
uint64_t getHostHash(void) {
char uname[1024];
// Start off with the hostname
(void) getHostName(uname, sizeof(uname));
int hlen = strlen(uname);
int len = readlink("/proc/self/ns/uts", uname+hlen, sizeof(uname)-1-hlen);
int offset = strlen(uname);
int len;
// $(readlink /proc/self/ns/uts)
len = readlink("/proc/self/ns/uts", uname+offset, sizeof(uname)-1-offset);
if (len < 0) len = 0;
uname[hlen+len]='\0';
offset += len;
// $(readlink /proc/self/ns/mnt)
len = readlink("/proc/self/ns/mnt", uname+offset, sizeof(uname)-1-offset);
if (len < 0) len = 0;
offset += len;
// Trailing '\0'
uname[offset]='\0';
TRACE(INIT,"unique hostname '%s'", uname);
return getHash(uname);