From 39fadb9b0446839a3a87e15bef15526438a6e792 Mon Sep 17 00:00:00 2001
From: Brian Kelley <bmkelle@sandia.gov>
Date: Fri, 4 Feb 2022 13:26:53 -0700
Subject: [PATCH] Fix 1303: maintain correct #cols on A in twostage

---
 src/sparse/impl/KokkosSparse_twostage_gauss_seidel_impl.hpp | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/sparse/impl/KokkosSparse_twostage_gauss_seidel_impl.hpp b/src/sparse/impl/KokkosSparse_twostage_gauss_seidel_impl.hpp
index 35c9fd3af7..19bc5ec163 100644
--- a/src/sparse/impl/KokkosSparse_twostage_gauss_seidel_impl.hpp
+++ b/src/sparse/impl/KokkosSparse_twostage_gauss_seidel_impl.hpp
@@ -916,8 +916,8 @@ class TwostageGaussSeidel {
         gsHandle->getUa();  // complement of U+D (used only for compact form)
 
     // wratp A into crsmat
-    input_graph_t graphA(column_view, rowmap_view);
-    input_crsmat_t crsmatA("A", num_rows, values_view, graphA);
+    input_crsmat_t crsmatA("A", num_rows, num_cols, values_view.extent(0),
+                           values_view, rowmap_view, column_view);
 #ifdef KOKKOSSPARSE_IMPL_TIME_TWOSTAGE_GS
     Kokkos::fence();
     tic = timer.seconds();