diff --git a/src/batched/KokkosBatched_InverseLU_Serial_Impl.hpp b/src/batched/KokkosBatched_InverseLU_Serial_Impl.hpp index 0615e928c9..8a815de407 100644 --- a/src/batched/KokkosBatched_InverseLU_Serial_Impl.hpp +++ b/src/batched/KokkosBatched_InverseLU_Serial_Impl.hpp @@ -88,7 +88,7 @@ namespace KokkosBatched { auto B = Kokkos::View >(W.data(), A.extent(0), A.extent(1)); const ScalarType one(1.0); - + #if defined(KOKKOS_ENABLE_PRAGMA_UNROLL) #pragma unroll #endif diff --git a/src/batched/KokkosBatched_SolveLU_Serial_Impl.hpp b/src/batched/KokkosBatched_SolveLU_Serial_Impl.hpp index 78fdc2a01c..1a5e18c369 100644 --- a/src/batched/KokkosBatched_SolveLU_Serial_Impl.hpp +++ b/src/batched/KokkosBatched_SolveLU_Serial_Impl.hpp @@ -40,7 +40,7 @@ namespace KokkosBatched { SerialTrsm::invoke(one, A, B); //Second, compute X by solving the system U*X = Y for X SerialTrsm::invoke(one, A, B); - + return 0; } diff --git a/src/batched/KokkosBatched_Trsm_Serial_Internal.hpp b/src/batched/KokkosBatched_Trsm_Serial_Internal.hpp index 00bbe1948e..ec344309ce 100644 --- a/src/batched/KokkosBatched_Trsm_Serial_Internal.hpp +++ b/src/batched/KokkosBatched_Trsm_Serial_Internal.hpp @@ -180,8 +180,8 @@ namespace KokkosBatched { const int iend = p, jend = n; const ValueType *__restrict__ a01 = A+p*as1; - /**/ ValueType *__restrict__ b1t = B+p*bs0; - + ValueType *__restrict__ b1t = B+p*bs0; + if (!use_unit_diag) { const ValueType alpha11 = A[p*as0+p*as1]; @@ -191,13 +191,16 @@ namespace KokkosBatched { for (int j=0;j0){ + for (int i=0;i