NASP/cuda-samples
47
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2025-07-01 20:20:29 +08:00
Code Issues Packages Projects Releases Wiki Activity

Bug 5207005: Append pid in shmName for Linux only as this is for MIG scenario

Browse Source
This commit is contained in:
shawnz 2025-04-07 17:17:17 +08:00
parent ac700327a2
commit e77d6eb5ab
2 changed files with 20 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Samples/0_Introduction/simpleIPC/simpleIPC.cu
View File

@ -99,12 +99,15 @@ static void childProcess(int id)
std::vector<void *> ptrs; std::vector<void *> ptrs;
std::vector<cudaEvent_t> events; std::vector<cudaEvent_t> events;
std::vector<char> verification_buffer(DATA_SIZE); std::vector<char> verification_buffer(DATA_SIZE);
pid_t pid;
char pidString[20] = {0}; char pidString[20] = {0};
char lshmName[40] = {0}; char lshmName[40] = {0};
// Use parent process ID to create a unique shared memory name for Linux multi-process
#ifdef __linux__
pid_t pid;
pid = getppid(); pid = getppid();
snprintf(pidString, sizeof(pidString), "%d", pid); snprintf(pidString, sizeof(pidString), "%d", pid);
#endif
strcat(lshmName, shmName); strcat(lshmName, shmName);
strcat(lshmName, pidString); strcat(lshmName, pidString);
@ -205,12 +208,15 @@ static void parentProcess(char *app)
std::vector<void *> ptrs; std::vector<void *> ptrs;
std::vector<cudaEvent_t> events; std::vector<cudaEvent_t> events;
std::vector<Process> processes; std::vector<Process> processes;
pid_t pid;
char pidString[20] = {0}; char pidString[20] = {0};
char lshmName[40] = {0}; char lshmName[40] = {0};
// Use current process ID to create a unique shared memory name for Linux multi-process
#ifdef __linux__
pid_t pid;
pid = getpid(); pid = getpid();
snprintf(pidString, sizeof(pidString), "%d", pid); snprintf(pidString, sizeof(pidString), "%d", pid);
#endif
strcat(lshmName, shmName); strcat(lshmName, shmName);
strcat(lshmName, pidString); strcat(lshmName, pidString);

16
Samples/3_CUDA_Features/memMapIPCDrv/memMapIpc.cpp
View File

@ -310,12 +310,16 @@ static void childProcess(int devId, int id, char **argv)
ipcHandle *ipcChildHandle = NULL; ipcHandle *ipcChildHandle = NULL;
int blocks = 0; int blocks = 0;
int threads = 128; int threads = 128;
pid_t pid; char pidString[20] = {0};
char pidString[20] = {0}; char lshmName[40] = {0};
char lshmName[40] = {0};
// Use parent process ID to create a unique shared memory name for Linux multi-process
#ifdef __linux__
pid_t pid;
pid = getppid(); pid = getppid();
snprintf(pidString, sizeof(pidString), "%d", pid); snprintf(pidString, sizeof(pidString), "%d", pid);
#endif
strcat(lshmName, shmName); strcat(lshmName, shmName);
strcat(lshmName, pidString); strcat(lshmName, pidString);
@ -431,16 +435,20 @@ static void parentProcess(char *app)
volatile shmStruct *shm = NULL; volatile shmStruct *shm = NULL;
sharedMemoryInfo info; sharedMemoryInfo info;
std::vector<Process> processes; std::vector<Process> processes;
pid_t pid;
char pidString[20] = {0}; char pidString[20] = {0};
char lshmName[40] = {0}; char lshmName[40] = {0};
// Use current process ID to create a unique shared memory name for Linux multi-process
#ifdef __linux__
pid_t pid;
pid = getpid(); pid = getpid();
snprintf(pidString, sizeof(pidString), "%d", pid); snprintf(pidString, sizeof(pidString), "%d", pid);
#endif
strcat(lshmName, shmName); strcat(lshmName, shmName);
strcat(lshmName, pidString); strcat(lshmName, pidString);
printf("PP: lshmName = %s\n", lshmName); printf("PP: lshmName = %s\n", lshmName);
checkCudaErrors(cuDeviceGetCount(&devCount)); checkCudaErrors(cuDeviceGetCount(&devCount));
std::vector<CUdevice> devices(devCount); std::vector<CUdevice> devices(devCount);