Resolve a bug for the Batched GMRES with RightLayout

kokkos · Nov 16, 2021 · 21db7c1 · 21db7c1
1 parent 9ee4836
commit 21db7c1
Show file tree

Hide file tree

Showing 4 changed files with 6 additions and 6 deletions.
diff --git a/src/batched/sparse/KokkosBatched_CG.hpp b/src/batched/sparse/KokkosBatched_CG.hpp
@@ -49,14 +49,14 @@
 
 /// \brief Batched CG: Selective Interface
 ///
-/// \tparam OperatorType: The type of the opertator of the system
+/// \tparam OperatorType: The type of the operator of the system
 /// \tparam VectorViewType: Input type for the right-hand side and the solution,
 /// needs to be a 2D view
 ///
 /// \param member [in]: TeamPolicy member
 /// \param A [in]: batched operator (can be a batched matrix or a (left or right
 /// or both) preconditioned batched matrix) \param B [in]: right-hand side, a
-/// rank 2 view \param X [in/out]: initial guess and solutin, a rank 2 view
+/// rank 2 view \param X [in/out]: initial guess and solution, a rank 2 view
 /// \param handle [in]: a handle which provides different information such as
 /// the tolerance or the maximal number of iterations of the solver.
 

diff --git a/src/batched/sparse/KokkosBatched_GMRES.hpp b/src/batched/sparse/KokkosBatched_GMRES.hpp
@@ -49,14 +49,14 @@
 
 /// \brief Batched GMRES: Selective Interface
 ///
-/// \tparam OperatorType: The type of the opertator of the system
+/// \tparam OperatorType: The type of the operator of the system
 /// \tparam VectorViewType: Input type for the right-hand side and the solution,
 /// needs to be a 2D view
 ///
 /// \param member [in]: TeamPolicy member
 /// \param A [in]: batched operator (can be a batched matrix or a (left or right
 /// or both) preconditioned batched matrix) \param B [in]: right-hand side, a
-/// rank 2 view \param X [in/out]: initial guess and solutin, a rank 2 view
+/// rank 2 view \param X [in/out]: initial guess and solution, a rank 2 view
 /// \param handle [in]: a handle which provides different information such as
 /// the tolerance or the maximal number of iterations of the solver.
 

diff --git a/src/batched/sparse/impl/KokkosBatched_GMRES_TeamVector_Impl.hpp b/src/batched/sparse/impl/KokkosBatched_GMRES_TeamVector_Impl.hpp
@@ -103,7 +103,7 @@ struct TeamVectorGMRES {
     ScratchPadMultiVectorViewType Givens(member.team_scratch(1), numMatrices,
                                          maximum_iteration, 2);
     ScratchPadVectorViewType G(member.team_scratch(1), numMatrices,
-                               maximum_iteration);
+                               maximum_iteration + 1);
 
     ScratchPadVectorViewType W(member.team_scratch(0), numMatrices, numRows);
     ScratchPadVectorViewType Q(member.team_scratch(0), numMatrices, numRows);

diff --git a/src/batched/sparse/impl/KokkosBatched_GMRES_Team_Impl.hpp b/src/batched/sparse/impl/KokkosBatched_GMRES_Team_Impl.hpp
@@ -102,7 +102,7 @@ struct TeamGMRES {
     ScratchPadMultiVectorViewType Givens(member.team_scratch(1), numMatrices,
                                          maximum_iteration, 2);
     ScratchPadVectorViewType G(member.team_scratch(1), numMatrices,
-                               maximum_iteration);
+                               maximum_iteration + 1);
 
     ScratchPadVectorViewType W(member.team_scratch(0), numMatrices, numRows);
     ScratchPadVectorViewType Q(member.team_scratch(0), numMatrices, numRows);