floraachy
/
OpenBLAS
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Merge pull request #1354 from martin-frbg/shmem 

Try to handle shmget or shmat failing
This commit is contained in:
Martin Kroeker 2017-11-10 09:11:03 +01:00 committed by GitHub
parent cc26cdce0c 148493df89
commit 4271b2b158
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 32 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

46
driver/others/init.c
View File

@ -631,7 +631,7 @@ static inline int is_dead(int id) {
return shmctl(id, IPC_STAT, &ds);
}
static void open_shmem(void) {
static int open_shmem(void) {
int try = 0;
@ -657,30 +657,42 @@ static void open_shmem(void) {
} while ((try < 10) && (shmid == -1));
if (shmid == -1) {
fprintf(stderr, "GotoBLAS : Can't open shared memory. Terminated.\n");
exit(1);
perror ("Obtaining shared memory segment failed in open_shmem");
return (1);
}
if (shmid != -1) {
if ( (common = shmat(shmid, NULL, 0)) == (void*)-1) perror ("Attaching shared memory segment");
if ( (common = shmat(shmid, NULL, 0)) == (void*)-1) {
perror ("Attaching shared memory segment failed in open_shmem");
return (1);
}
}
#ifdef DEBUG
fprintf(stderr, "Shared Memory id = %x Address = %p\n", shmid, common);
#endif
return (0);
}
static void create_pshmem(void) {
static int create_pshmem(void) {
pshmid = shmget(IPC_PRIVATE, 4096, IPC_CREAT | 0666);
paddr = shmat(pshmid, NULL, 0);
shmctl(pshmid, IPC_RMID, 0);
if (pshmid == -1) {
perror ("Obtaining shared memory segment failed in create_pshmem");
return(1);
}
if ( (paddr = shmat(pshmid, NULL, 0)) == (void*)-1) {
perror ("Attaching shared memory segment failed in create_pshmem");
return (1);
}
if (shmctl(pshmid, IPC_RMID, 0) == -1) return (1);
#ifdef DEBUG
fprintf(stderr, "Private Shared Memory id = %x Address = %p\n", pshmid, paddr);
#endif
return(0);
}
static void local_cpu_map(void) {
@ -808,17 +820,23 @@ void gotoblas_affinity_init(void) {
return;
}
create_pshmem();
open_shmem();
if (create_pshmem() != 0) {
disable_mapping = 1;
return;
}
if (open_shmem() != 0) {
disable_mapping = 1;
return;
}
while ((common -> lock) && (common -> magic != SH_MAGIC)) {
if (is_dead(common -> shmid)) {
common -> lock = 0;
common -> shmid = 0;
common -> magic = 0;
} else {
sched_yield();
YIELDING;
}
}