From 6b0848ceafa5e1b24ed4a5d30d76fd904fe96774 Mon Sep 17 00:00:00 2001 From: Akarshan Biswas Date: Thu, 12 Dec 2024 19:44:06 +0530 Subject: [PATCH] SYCL ggml-sycl: pool2D use sycl::nan and remove if-else block --- ggml/src/ggml-sycl/ggml-sycl.cpp | 12 +++--------- 1 file changed, 3 insertions(+), 9 deletions(-) diff --git a/ggml/src/ggml-sycl/ggml-sycl.cpp b/ggml/src/ggml-sycl/ggml-sycl.cpp index a353309e4..6b9f0b0d9 100644 --- a/ggml/src/ggml-sycl/ggml-sycl.cpp +++ b/ggml/src/ggml-sycl/ggml-sycl.cpp @@ -1789,14 +1789,12 @@ static void pool2d_nchw_kernel( const int ew = sycl::min(iw, start_w + kw); To res = 0; - bool op_valid = true; switch (op) { case GGML_OP_POOL_AVG: res = 0; break; case GGML_OP_POOL_MAX: res = -FLT_MAX; break; default: - res = NAN; - op_valid = false; + res = (To) sycl::nan(uint32_t(0)); break; } @@ -1817,16 +1815,12 @@ static void pool2d_nchw_kernel( case GGML_OP_POOL_AVG: res += (cur / (kh * kw)); break; case GGML_OP_POOL_MAX: res = sycl::max(res, (To)cur); break; default: - op_valid = false; + res = (To) sycl::nan(uint32_t(0)); break; } } } - if (op_valid) { - o_ptr[cur_oh * ow + cur_ow] = res; - } else { - o_ptr[cur_oh * ow + cur_ow] = NAN; - } + o_ptr[cur_oh * ow + cur_ow] = res; } template