diff --git a/interface/gemm_batch.c b/interface/gemm_batch.c index b0d5b29cc..0d8854409 100644 --- a/interface/gemm_batch.c +++ b/interface/gemm_batch.c @@ -121,8 +121,8 @@ void NAME(char *transa_array, char *transb_array, IFLOAT ** a_array, blasint * lda_array, IFLOAT ** b_array, blasint * ldb_array, FLOAT * beta_array, - FLOAT ** c_array, blasint * ldc_array, blasint group_count, blasint * group_size) { - + FLOAT ** c_array, blasint * ldc_array, blasint * gcount, blasint * group_size) { + blasint group_count = *gcount; #else void CNAME(enum CBLAS_ORDER order, enum CBLAS_TRANSPOSE * transa_array, enum CBLAS_TRANSPOSE * transb_array,