avoid a potential division by zero

This commit is contained in:
Martin Kroeker 2024-06-27 11:31:19 +02:00 committed by GitHub
parent 1ba1b9c357
commit 831ef0d5d5
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 2 additions and 0 deletions

View File

@ -705,6 +705,7 @@ static int gemm_driver(blas_arg_t *args, BLASLONG *range_m, BLASLONG
range_M[num_parts + 1] = range_M[num_parts] + width; range_M[num_parts + 1] = range_M[num_parts] + width;
num_parts ++; num_parts ++;
if (num_parts == nthreads_m) break;
} }
for (i = num_parts; i < MAX_CPU_NUMBER; i++) { for (i = num_parts; i < MAX_CPU_NUMBER; i++) {
range_M[i + 1] = range_M[num_parts]; range_M[i + 1] = range_M[num_parts];
@ -752,6 +753,7 @@ static int gemm_driver(blas_arg_t *args, BLASLONG *range_m, BLASLONG
range_N[num_parts + 1] = range_N[num_parts] + width; range_N[num_parts + 1] = range_N[num_parts] + width;
num_parts ++; num_parts ++;
if (num_parts == nthreads) break;
} }
for (j = num_parts; j < MAX_CPU_NUMBER; j++) { for (j = num_parts; j < MAX_CPU_NUMBER; j++) {
range_N[j + 1] = range_N[num_parts]; range_N[j + 1] = range_N[num_parts];