diff options
| author | czgdp1807 <gdp.1807@gmail.com> | 2021-06-09 10:02:50 +0530 |
|---|---|---|
| committer | czgdp1807 <gdp.1807@gmail.com> | 2021-06-09 10:02:50 +0530 |
| commit | 3f1c88c13a28c9b54f40d1b474efc2c3aee49c8c (patch) | |
| tree | 8ed14db3f61e14396743230b57bf524dfc2441e8 /numpy/linalg/umath_linalg.c.src | |
| parent | 03bfafc427f23f3335f41d541730df6e9cfc514c (diff) | |
| download | numpy-3f1c88c13a28c9b54f40d1b474efc2c3aee49c8c.tar.gz | |
updated method for finding optimal work_count
Diffstat (limited to 'numpy/linalg/umath_linalg.c.src')
| -rw-r--r-- | numpy/linalg/umath_linalg.c.src | 30 |
1 files changed, 21 insertions, 9 deletions
diff --git a/numpy/linalg/umath_linalg.c.src b/numpy/linalg/umath_linalg.c.src index 3d74db210..070d41929 100644 --- a/numpy/linalg/umath_linalg.c.src +++ b/numpy/linalg/umath_linalg.c.src @@ -2984,7 +2984,7 @@ init_@lapack_func@(GEQRF_PARAMS_t *params, } params->LWORK = fortran_int_max(fortran_int_max(1, n), - work_count); + work_count > 0 ? work_count : -work_count); work_size = (size_t) params->LWORK * sizeof(@ftyp@); mem_buff2 = malloc(work_size); @@ -3029,7 +3029,7 @@ init_@lapack_func@(GEQRF_PARAMS_t *params, size_t a_size = safe_m * safe_n * sizeof(@ftyp@); size_t tau_size = safe_min_m_n * sizeof(@ftyp@); - fortran_int work_count; + fortran_int work_count_r, work_count_i, work_count; size_t work_size; fortran_int lda = fortran_int_max(1, m); @@ -3060,7 +3060,11 @@ init_@lapack_func@(GEQRF_PARAMS_t *params, if (call_@lapack_func@(params) != 0) goto error; - work_count = (fortran_int) ((@ftyp@*)params->WORK)->r; + work_count_r = (fortran_int) ((@ftyp@*)params->WORK)->r; + work_count_i = (fortran_int) ((@ftyp@*)params->WORK)->i; + + work_count = (fortran_int) sqrt((fortran_doublereal) work_count_r*work_count_r + + (fortran_doublereal) work_count_i*work_count_i); } @@ -3276,7 +3280,7 @@ init_@lapack_func@(GQR_PARAMS_t *params, } params->LWORK = fortran_int_max(fortran_int_max(1, n), - work_count); + work_count > 0 ? work_count : -work_count); work_size = (size_t) params->LWORK * sizeof(@ftyp@); @@ -3323,7 +3327,7 @@ init_@lapack_func@(GQR_PARAMS_t *params, size_t q_size = safe_m * safe_min_m_n * sizeof(@ftyp@); size_t tau_size = safe_min_m_n * sizeof(@ftyp@); - fortran_int work_count; + fortran_int work_count_r, work_count_i, work_count; size_t work_size; fortran_int lda = fortran_int_max(1, m); @@ -3355,7 +3359,11 @@ init_@lapack_func@(GQR_PARAMS_t *params, if (call_@lapack_func@(params) != 0) goto error; - work_count = (fortran_int) ((@ftyp@*)params->WORK)->r; + work_count_r = (fortran_int) ((@ftyp@*)params->WORK)->r; + work_count_i = (fortran_int) ((@ftyp@*)params->WORK)->i; + + work_count = (fortran_int) sqrt((fortran_doublereal) work_count_r*work_count_r + + (fortran_doublereal) work_count_i*work_count_i); } @@ -3519,7 +3527,7 @@ init_@lapack_func@_complete(GQR_PARAMS_t *params, } params->LWORK = fortran_int_max(fortran_int_max(1, n), - work_count); + work_count > 0 ? work_count : -work_count); work_size = (size_t) params->LWORK * sizeof(@ftyp@); @@ -3566,7 +3574,7 @@ init_@lapack_func@_complete(GQR_PARAMS_t *params, size_t q_size = safe_m * safe_m * sizeof(@ftyp@); size_t tau_size = safe_min_m_n * sizeof(@ftyp@); - fortran_int work_count; + fortran_int work_count_r, work_count_i, work_count; size_t work_size; fortran_int lda = fortran_int_max(1, m); @@ -3598,7 +3606,11 @@ init_@lapack_func@_complete(GQR_PARAMS_t *params, if (call_@lapack_func@(params) != 0) goto error; - work_count = (fortran_int) ((@ftyp@*)params->WORK)->r; + work_count_r = (fortran_int) ((@ftyp@*)params->WORK)->r; + work_count_i = (fortran_int) ((@ftyp@*)params->WORK)->i; + + work_count = (fortran_int) sqrt((fortran_doublereal) work_count_r*work_count_r + + (fortran_doublereal) work_count_i*work_count_i); } |