Skip to content

Commit

Permalink
modify the blas_l1_thread.c for support multi-threded for L1 fuction …
Browse files Browse the repository at this point in the history
…with return value
  • Loading branch information
jiahaipeng authored and xianyi committed Jan 10, 2017
1 parent cbd2bf1 commit 1aa1e6c
Showing 1 changed file with 71 additions and 0 deletions.
71 changes: 71 additions & 0 deletions driver/others/blas_l1_thread.c
Original file line number Diff line number Diff line change
Expand Up @@ -110,3 +110,74 @@ int blas_level1_thread(int mode, BLASLONG m, BLASLONG n, BLASLONG k, void *alpha

return 0;
}

int blas_level1_thread_with_return_value(int mode, BLASLONG m, BLASLONG n, BLASLONG k, void *alpha,
void *a, BLASLONG lda,
void *b, BLASLONG ldb,
void *c, BLASLONG ldc, int (*function)(), int nthreads){

blas_queue_t queue[MAX_CPU_NUMBER];
blas_arg_t args [MAX_CPU_NUMBER];

BLASLONG i, width, astride, bstride;
int num_cpu, calc_type;

calc_type = (mode & BLAS_PREC) + ((mode & BLAS_COMPLEX) != 0) + 2;

mode |= BLAS_LEGACY;

for (i = 0; i < nthreads; i++) blas_queue_init(&queue[i]);

num_cpu = 0;
i = m;

while (i > 0){

/* Adjust Parameters */
width = blas_quickdivide(i + nthreads - num_cpu - 1,
nthreads - num_cpu);

i -= width;
if (i < 0) width = width + i;

astride = width * lda;

if (!(mode & BLAS_TRANSB_T)) {
bstride = width * ldb;
} else {
bstride = width;
}

astride <<= calc_type;
bstride <<= calc_type;

args[num_cpu].m = width;
args[num_cpu].n = n;
args[num_cpu].k = k;
args[num_cpu].a = (void *)a;
args[num_cpu].b = (void *)b;
args[num_cpu].c = (void *)((char *)c + num_cpu * sizeof(double)*2);
args[num_cpu].lda = lda;
args[num_cpu].ldb = ldb;
args[num_cpu].ldc = ldc;
args[num_cpu].alpha = alpha;

queue[num_cpu].mode = mode;
queue[num_cpu].routine = function;
queue[num_cpu].args = &args[num_cpu];
queue[num_cpu].next = &queue[num_cpu + 1];

a = (void *)((BLASULONG)a + astride);
b = (void *)((BLASULONG)b + bstride);

num_cpu ++;
}

if (num_cpu) {
queue[num_cpu - 1].next = NULL;

exec_blas(num_cpu, queue);
}

return 0;
}

0 comments on commit 1aa1e6c

Please sign in to comment.