From d0877f4f20bc2fbf5a488c620d74ae634dd992c1 Mon Sep 17 00:00:00 2001 From: Feng Yuan Date: Mon, 4 Mar 2024 21:06:03 +0800 Subject: [PATCH] Loops: align vectorization code-path with queried vectorization size Signed-off-by: Feng Yuan --- src/aten/sycl/Loops.h | 3 +++ 1 file changed, 3 insertions(+) diff --git a/src/aten/sycl/Loops.h b/src/aten/sycl/Loops.h index 634afc0ad..4c571a6fb 100644 --- a/src/aten/sycl/Loops.h +++ b/src/aten/sycl/Loops.h @@ -351,6 +351,9 @@ static inline void launch_vectorized_kernel( } switch (vec_size) { + case 16: + VEC_KER(16); + break; case 8: VEC_KER(8); break;