maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

Insert missing checks for KMP_AFFINITY_CAPABLE() in affinity API. 

If affinity is not capable, then these API functions will perform the stubs
version.

llvm-svn: 282947
This commit is contained in:
Jonathan Peyton 2016-09-30 20:56:44 +00:00
parent 129795497d
commit 66e212ce2b
1 changed files with 12 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
openmp/runtime/src/kmp_ftn_entry.h
View File

@ -690,6 +690,8 @@ FTN_GET_NUM_PLACES( void )
if ( ! TCR_4(__kmp_init_middle) ) {
__kmp_middle_initialize();
}
if (!KMP_AFFINITY_CAPABLE())
return 0;
return __kmp_affinity_num_masks;
#endif
}
@ -705,6 +707,8 @@ FTN_GET_PLACE_NUM_PROCS( int place_num )
if ( ! TCR_4(__kmp_init_middle) ) {
__kmp_middle_initialize();
}
if (!KMP_AFFINITY_CAPABLE())
return 0;
if ( place_num < 0 || place_num >= (int)__kmp_affinity_num_masks )
return 0;
kmp_affin_mask_t *mask = KMP_CPU_INDEX(__kmp_affinity_masks, place_num);
@ -729,6 +733,8 @@ FTN_GET_PLACE_PROC_IDS( int place_num, int *ids )
if ( ! TCR_4(__kmp_init_middle) ) {
__kmp_middle_initialize();
}
if (!KMP_AFFINITY_CAPABLE())
return;
if ( place_num < 0 || place_num >= (int)__kmp_affinity_num_masks )
return;
kmp_affin_mask_t *mask = KMP_CPU_INDEX(__kmp_affinity_masks, place_num);
@ -754,6 +760,8 @@ FTN_GET_PLACE_NUM( void )
if ( ! TCR_4(__kmp_init_middle) ) {
__kmp_middle_initialize();
}
if (!KMP_AFFINITY_CAPABLE())
return -1;
gtid = __kmp_entry_gtid();
thread = __kmp_thread_from_gtid(gtid);
if ( thread->th.th_current_place < 0 )
@ -773,6 +781,8 @@ FTN_GET_PARTITION_NUM_PLACES( void )
if ( ! TCR_4(__kmp_init_middle) ) {
__kmp_middle_initialize();
}
if (!KMP_AFFINITY_CAPABLE())
return 0;
gtid = __kmp_entry_gtid();
thread = __kmp_thread_from_gtid(gtid);
first_place = thread->th.th_first_place;
@ -797,6 +807,8 @@ FTN_GET_PARTITION_PLACE_NUMS( int *place_nums ) {
if ( ! TCR_4(__kmp_init_middle) ) {
__kmp_middle_initialize();
}
if (!KMP_AFFINITY_CAPABLE())
return;
gtid = __kmp_entry_gtid();
thread = __kmp_thread_from_gtid(gtid);
first_place = thread->th.th_first_place;