Browse Source

Fix range limit exceeding data size in last step

tags/v0.3.0
Martin Kroeker GitHub 8 years ago
parent
commit
857f61bc5d
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      driver/level2/sbmv_thread.c

+ 2
- 0
driver/level2/sbmv_thread.c View File

@@ -246,6 +246,7 @@ int CNAME(BLASLONG n, BLASLONG k, FLOAT *alpha, FLOAT *a, BLASLONG lda, FLOAT *x

range_m[MAX_CPU_NUMBER - num_cpu - 1] = range_m[MAX_CPU_NUMBER - num_cpu] - width;
range_n[num_cpu] = num_cpu * (((n + 15) & ~15) + 16);
if (range_n[num_cpu] > n) range_n[num_cpu] = n;

queue[num_cpu].mode = mode;
queue[num_cpu].routine = sbmv_kernel;
@@ -285,6 +286,7 @@ int CNAME(BLASLONG n, BLASLONG k, FLOAT *alpha, FLOAT *a, BLASLONG lda, FLOAT *x

range_m[num_cpu + 1] = range_m[num_cpu] + width;
range_n[num_cpu] = num_cpu * (((n + 15) & ~15) + 16);
if (range_n[num_cpu] > n) range_n[num_cpu] = n;

queue[num_cpu].mode = mode;
queue[num_cpu].routine = sbmv_kernel;


Loading…
Cancel
Save