From 6713c3677352bd7c46bef55cc5b5629e2b7b78f8 Mon Sep 17 00:00:00 2001
From: Luc Berger-Vergiat <lberge@sandia.gov>
Date: Thu, 6 Mar 2025 08:52:44 -0700
Subject: [PATCH] Fixing issue with pragma unroll in laswp

Signed-off-by: Luc Berger-Vergiat <lberge@sandia.gov>
---
 batched/dense/impl/KokkosBatched_Laswp_Serial_Internal.hpp | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/batched/dense/impl/KokkosBatched_Laswp_Serial_Internal.hpp b/batched/dense/impl/KokkosBatched_Laswp_Serial_Internal.hpp
index fd0056ddf8..d02bfd96f3 100644
--- a/batched/dense/impl/KokkosBatched_Laswp_Serial_Internal.hpp
+++ b/batched/dense/impl/KokkosBatched_Laswp_Serial_Internal.hpp
@@ -97,10 +97,12 @@ struct SerialLaswpVectorBackwardInternal {
 // On H100 with Cuda 12.0.0, the compiler seems to apply
 // an aggressive optimization which crashes this function
 // Disabling loop unrolling fixes the issue
+#if defined(KOKKOS_ENABLE_PRAGMA_UNROLL)
 #if defined(KOKKOS_ENABLE_CUDA) && defined(KOKKOS_ARCH_HOPPER90)
 #if CUDA_VERSION >= 12000 && CUDA_VERSION < 12100
 #pragma unroll 1
 #endif
+#endif
 #endif
     for (int i = (plen - 1); i >= 0; --i) {
       const int piv = p[i * ps0];