Skip to content

Commit f216847

Browse files
authored
fix type confusions
1 parent 9f94682 commit f216847

1 file changed

Lines changed: 15 additions & 8 deletions

File tree

interface/gemm_batch_strided.c

Lines changed: 15 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -117,12 +117,12 @@ static size_t zgemm_small_kernel_b0[] = {
117117
#ifndef CBLAS
118118
void NAME(char *transa, char *transb,
119119
blasint * M, blasint * N, blasint * K,
120-
FLOAT * alpha,
120+
FLOAT * Alpha,
121121
IFLOAT * a, blasint * Lda,
122122
blasint * stride_a,
123123
IFLOAT *b, blasint * Ldb,
124124
blasint * stride_b,
125-
FLOAT * beta,
125+
FLOAT * Beta,
126126
FLOAT * c, blasint * Ldc, blasint * stride_c, blasint * matcount) {
127127

128128
char ta = *transa;
@@ -137,15 +137,22 @@ void NAME(char *transa, char *transb,
137137
blasint lda=*Lda;
138138
blasint ldb=*Ldb;
139139
blasint ldc=*Ldc;
140+
#if !defined(COMPLEX)
141+
FLOAT alpha=*Alpha;
142+
FLOAT beta=*Beta;
143+
#else
144+
FLOAT *alpha=Alpha;
145+
FLOAT *beta=Beta;
146+
#endif
140147
#else
141148

142149
void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE transa, enum CBLAS_TRANSPOSE transb,
143150
blasint m, blasint n, blasint k,
144151
#ifndef COMPLEX
145-
FLOAT * alpha,
152+
FLOAT alpha,
146153
IFLOAT * a, blasint lda, blasint stridea,
147154
IFLOAT * b, blasint ldb, blasint strideb,
148-
FLOAT * beta,
155+
FLOAT beta,
149156
FLOAT * c, blasint ldc, blasint stridec, blasint count) {
150157
#else
151158
void * valpha,
@@ -352,15 +359,15 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE transa, enum CBLAS_TRANS
352359
if (MNK <= 100.0*100.0*100.0){
353360
group_routine=NULL;
354361
#if !defined(COMPLEX)
355-
if(*(FLOAT *)(beta) == 0.0){
362+
if(beta == 0.0){
356363
group_mode=mode | BLAS_SMALL_B0_OPT;
357364
group_small_matrix_opt_routine=(void *)(gemm_small_kernel_b0[(group_transb<<2)|group_transa]);
358365
}else{
359366
group_mode=mode | BLAS_SMALL_OPT;
360367
group_small_matrix_opt_routine=(void *)(gemm_small_kernel[(group_transb<<2)|group_transa]);
361368
}
362369
#else
363-
if(((FLOAT *)(beta))[0] == 0.0 && ((FLOAT *)(beta))[1] == 0.0){
370+
if(beta[0] == 0.0 && beta[1] == 0.0){
364371
group_mode=mode | BLAS_SMALL_B0_OPT;
365372
group_small_matrix_opt_routine=(void *)(zgemm_small_kernel_b0[(group_transb<<2)|group_transa]);
366373
}else{
@@ -384,8 +391,8 @@ void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE transa, enum CBLAS_TRANS
384391
args_array[i].lda=group_lda;
385392
args_array[i].ldb=group_ldb;
386393
args_array[i].ldc=group_ldc;
387-
args_array[i].alpha=alpha;
388-
args_array[i].beta=beta;
394+
args_array[i].alpha=&alpha;
395+
args_array[i].beta=&beta;
389396

390397
#if defined(CBLAS)
391398
if (order == CblasColMajor) {

0 commit comments

Comments
 (0)