softmax: remove pragma unroll directive
This commit is contained in:
parent
45d6c58dba
commit
1c5611ef1e
1 changed files with 0 additions and 1 deletions
|
@ -32,7 +32,6 @@ static void soft_max_f32(const float * x, const T * mask, float * dst, const int
|
||||||
float *vals = vals_smem ? buf + sycl::max(nwarps, WARP_SIZE) : dst + rowx * ncols;
|
float *vals = vals_smem ? buf + sycl::max(nwarps, WARP_SIZE) : dst + rowx * ncols;
|
||||||
float max_val = -INFINITY;
|
float max_val = -INFINITY;
|
||||||
|
|
||||||
#pragma unroll
|
|
||||||
for (int col0 = 0; col0 < ncols; col0 += block_size) {
|
for (int col0 = 0; col0 < ncols; col0 += block_size) {
|
||||||
const int col = col0 + tid;
|
const int col = col0 + tid;
|
||||||
|
|
||||||
|
|
Loading…
Add table
Add a link
Reference in a new issue