Update gemm3m_level3.c
authorwjc404 <52632443+wjc404@users.noreply.github.com>
Fri, 27 Dec 2019 10:03:01 +0000 (18:03 +0800)
committerGitHub <noreply@github.com>
Fri, 27 Dec 2019 10:03:01 +0000 (18:03 +0800)
driver/level3/gemm3m_level3.c

index bbde7e5..d037e72 100644 (file)
@@ -338,7 +338,7 @@ int CNAME(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n,
 
       for(jjs = js; jjs < js + min_j; jjs += min_jj){
        min_jj = min_j + js - jjs;
-       if (min_jj > GEMM3M_UNROLL_N) min_jj = GEMM3M_UNROLL_N;
+       if (min_jj > GEMM3M_UNROLL_N*3) min_jj = GEMM3M_UNROLL_N*3;
 
        START_RPCC();
 
@@ -398,7 +398,7 @@ int CNAME(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n,
 
       for(jjs = js; jjs < js + min_j; jjs += min_jj){
        min_jj = min_j + js - jjs;
-       if (min_jj > GEMM3M_UNROLL_N) min_jj = GEMM3M_UNROLL_N;
+       if (min_jj > GEMM3M_UNROLL_N*3) min_jj = GEMM3M_UNROLL_N*3;
 
        START_RPCC();
 
@@ -463,7 +463,7 @@ int CNAME(blas_arg_t *args, BLASLONG *range_m, BLASLONG *range_n,
 
       for(jjs = js; jjs < js + min_j; jjs += min_jj){
        min_jj = min_j + js - jjs;
-       if (min_jj > GEMM3M_UNROLL_N) min_jj = GEMM3M_UNROLL_N;
+       if (min_jj > GEMM3M_UNROLL_N*3) min_jj = GEMM3M_UNROLL_N*3;
 
        START_RPCC();