Browse Source

Use m or n as hard limit in range check depending on TRANSA setting

pull/1261/head
Martin Kroeker GitHub 8 years ago
parent
commit
6e753c5a71
1 changed files with 2 additions and 1 deletions
  1. +2
    -1
      driver/level2/gbmv_thread.c

+ 2
- 1
driver/level2/gbmv_thread.c View File

@@ -230,10 +230,11 @@ int CNAME(BLASLONG m, BLASLONG n, BLASLONG ku, BLASLONG kl, FLOAT *alpha, FLOAT

#ifndef TRANSA
range_m[num_cpu] = num_cpu * ((m + 15) & ~15);
if (range_m[num_cpu] > m) range_m[num_cpu] = m;
#else
range_m[num_cpu] = num_cpu * ((n + 15) & ~15);
#endif
if (range_m[num_cpu] > n) range_m[num_cpu] = n;
#endif

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


Loading…
Cancel
Save