Skip to content

Commit 3326b92

Browse files
authored
remove status variable blas_num_threads_set; initialize openmp thread maximum on startup
1 parent ea669c8 commit 3326b92

File tree

3 files changed

+3
-6
lines changed

3 files changed

+3
-6
lines changed

driver/others/blas_server_omp.c

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,7 @@
6868
#endif
6969

7070
int blas_server_avail = 0;
71+
int blas_omp_number_max = 0;
7172

7273
extern int openblas_omp_adaptive_env();
7374

@@ -100,7 +101,6 @@ static void adjust_thread_buffers() {
100101

101102
void goto_set_num_threads(int num_threads) {
102103

103-
blas_num_threads_set = 1;
104104
if (num_threads < 0) blas_num_threads_set = 0;
105105
if (num_threads < 1) num_threads = blas_num_threads;
106106

@@ -125,6 +125,8 @@ void openblas_set_num_threads(int num_threads) {
125125
}
126126

127127
int blas_thread_init(void){
128+
if(blas_omp_number_max <= 0)
129+
blas_omp_number_max = omp_get_max_threads();
128130

129131
blas_get_cpu_number();
130132

driver/others/memory.c

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -422,8 +422,6 @@ This value is equal or large than blas_cpu_number. This means some threads are s
422422
*/
423423
int blas_num_threads = 0;
424424

425-
int blas_num_threads_set = 0;
426-
427425
int goto_get_num_procs (void) {
428426
return blas_cpu_number;
429427
}
@@ -1996,8 +1994,6 @@ This value is equal or large than blas_cpu_number. This means some threads are s
19961994
*/
19971995
int blas_num_threads = 0;
19981996

1999-
int blas_num_threads_set = 0;
2000-
20011997
int goto_get_num_procs (void) {
20021998
return blas_cpu_number;
20031999
}

driver/others/memory_qalloc.c

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -283,7 +283,6 @@ The numbers of threads in the thread pool.
283283
This value is equal or large than blas_cpu_number. This means some threads are sleep.
284284
*/
285285
int blas_num_threads = 0;
286-
int blas_num_threads_set = 0;
287286

288287
int goto_get_num_procs (void) {
289288
return blas_cpu_number;

0 commit comments

Comments
 (0)