mirror of
https://github.com/ggerganov/llama.cpp.git
synced 2025-01-30 13:53:03 +01:00
softmax: remove pragma unroll directive
This commit is contained in:
parent
45d6c58dba
commit
1c5611ef1e
@ -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…
Reference in New Issue
Block a user