softmax: remove pragma unroll directive

This commit is contained in:
Akarshan Biswas 2025-01-21 19:29:39 +05:30
parent 45d6c58dba
commit 1c5611ef1e
No known key found for this signature in database
GPG Key ID: 52A578A14B32134D

View File

@ -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 max_val = -INFINITY;
#pragma unroll
for (int col0 = 0; col0 < ncols; col0 += block_size) {
const int col = col0 + tid;