mirror of
https://github.com/NVIDIA/nccl-tests.git
synced 2026-10-06 00:47:37 +00:00
Unmesh Deodhar
2c52b4e910
Add a portable OS abstraction and CMake build entry points so the tests can build on Windows with MSVC and CUDA. Update the perf and verifiable sources to avoid POSIX-only APIs and compiler extensions where needed. Signed-off-by: Unmesh Deodhar <[email protected]>
72 lines
1.6 KiB
C++
72 lines
1.6 KiB
C++
/*************************************************************************
|
|
* Copyright (c) 2016-2025, NVIDIA CORPORATION. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#include "os.h"
|
|
|
|
#include <stdio.h>
|
|
#include <stdlib.h>
|
|
#include <string.h>
|
|
#include <strings.h>
|
|
|
|
#define HOSTID_FILE "/proc/sys/kernel/random/boot_id"
|
|
|
|
static uint64_t getHash(const char* string, size_t n) {
|
|
uint64_t result = 5381;
|
|
for (size_t c = 0; c < n; c++) {
|
|
result = ((result << 5) + result) ^ string[c];
|
|
}
|
|
return result;
|
|
}
|
|
|
|
uint64_t ncclTestGetHostHash(const char* hostname) {
|
|
char hostHash[1024];
|
|
|
|
snprintf(hostHash, sizeof(hostHash), "%s", hostname);
|
|
int offset = strlen(hostHash);
|
|
|
|
FILE *file = fopen(HOSTID_FILE, "r");
|
|
if (file != NULL) {
|
|
char *p;
|
|
if (fscanf(file, "%ms", &p) == 1) {
|
|
strncpy(hostHash+offset, p, sizeof(hostHash)-offset-1);
|
|
free(p);
|
|
}
|
|
fclose(file);
|
|
}
|
|
|
|
hostHash[sizeof(hostHash)-1] = '\0';
|
|
|
|
return getHash(hostHash, strlen(hostHash));
|
|
}
|
|
|
|
int ncclTestGetHostname(char* name, size_t len) {
|
|
return gethostname(name, len);
|
|
}
|
|
|
|
int ncclTestGetPid() {
|
|
return (int)getpid();
|
|
}
|
|
|
|
int ncclTestStrcasecmp(const char* s1, const char* s2) {
|
|
return strcasecmp(s1, s2);
|
|
}
|
|
|
|
int ncclTestStrncasecmp(const char* s1, const char* s2, size_t n) {
|
|
return strncasecmp(s1, s2, n);
|
|
}
|
|
|
|
int ncclTestAsprintf(char** strp, const char* fmt, ...) {
|
|
va_list args;
|
|
va_start(args, fmt);
|
|
int result = vasprintf(strp, fmt, args);
|
|
va_end(args);
|
|
return result;
|
|
}
|
|
|
|
void ncclTestSetlinebuf(FILE* stream) {
|
|
setlinebuf(stream);
|
|
}
|