From b2394a94464630e0c4715be6ce4dbe59bea92b79 Mon Sep 17 00:00:00 2001 From: Luc Berger-Vergiat Date: Tue, 25 Feb 2025 09:11:49 -0700 Subject: [PATCH 1/2] SYCL - SIMD: preventing AVX512 code to be compiled when targeting GPU There is a check missing to verify that we are not trying to compile AVX512 intrinsic in GPU code which is obviously not a good idea. Signed-off-by: Luc Berger-Vergiat --- batched/dense/src/KokkosBatched_Vector_SIMD.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/batched/dense/src/KokkosBatched_Vector_SIMD.hpp b/batched/dense/src/KokkosBatched_Vector_SIMD.hpp index 82c6e72cb6..1e0648fd2b 100644 --- a/batched/dense/src/KokkosBatched_Vector_SIMD.hpp +++ b/batched/dense/src/KokkosBatched_Vector_SIMD.hpp @@ -22,7 +22,7 @@ #include #include "KokkosKernels_Macros.hpp" -#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) +#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) || defined(SYCL_LANGUAGE_VERSION) #undef __KOKKOSBATCHED_ENABLE_AVX__ #else // compiler bug with AVX in some architectures From 3d4e37cbac38139c2763a58534092803a6ddf5f9 Mon Sep 17 00:00:00 2001 From: Luc Berger-Vergiat Date: Wed, 26 Feb 2025 11:38:05 -0700 Subject: [PATCH 2/2] Switching to a more appropriate compiler macro Signed-off-by: Luc Berger-Vergiat --- batched/dense/src/KokkosBatched_Vector_SIMD.hpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/batched/dense/src/KokkosBatched_Vector_SIMD.hpp b/batched/dense/src/KokkosBatched_Vector_SIMD.hpp index 1e0648fd2b..dd7b9d960a 100644 --- a/batched/dense/src/KokkosBatched_Vector_SIMD.hpp +++ b/batched/dense/src/KokkosBatched_Vector_SIMD.hpp @@ -22,7 +22,7 @@ #include #include "KokkosKernels_Macros.hpp" -#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) || defined(SYCL_LANGUAGE_VERSION) +#if defined(__CUDA_ARCH__) || defined(__HIP_DEVICE_COMPILE__) || defined(__SYCL_DEVICE_ONLY__) #undef __KOKKOSBATCHED_ENABLE_AVX__ #else // compiler bug with AVX in some architectures