From eca01ff14a5b133d5a680d0d3413cef3ddfb2e1c Mon Sep 17 00:00:00 2001
From: Christian Glusa <caglusa@sandia.gov>
Date: Thu, 8 Jun 2023 08:56:50 -0600
Subject: [PATCH] Zoltan2 Sphynx: Fix template args

---
 packages/zoltan2/sphynx/src/Zoltan2_Sphynx.hpp  |  2 +-
 .../test/sphynx/Sphynx_Research_Driver.cpp      | 17 ++++++++---------
 2 files changed, 9 insertions(+), 10 deletions(-)

diff --git a/packages/zoltan2/sphynx/src/Zoltan2_Sphynx.hpp b/packages/zoltan2/sphynx/src/Zoltan2_Sphynx.hpp
index 2b7e307a3a76..f778ba84aaf2 100644
--- a/packages/zoltan2/sphynx/src/Zoltan2_Sphynx.hpp
+++ b/packages/zoltan2/sphynx/src/Zoltan2_Sphynx.hpp
@@ -888,7 +888,7 @@ namespace Zoltan2 {
     Teuchos::RCP<lproblem_t> innerPolyProblem(new lproblem_t());
     innerPolyProblem->setOperator(laplacian_);
 
-    using btop_t = Belos::TpetraOperator<scalar_t>;
+    using btop_t = Belos::TpetraOperator<scalar_t, lno_t, gno_t, node_t>;
     Teuchos::RCP<btop_t> polySolver(new btop_t(innerPolyProblem,
           Teuchos::rcpFromRef(paramList),
           "GmresPoly", true));
diff --git a/packages/zoltan2/test/sphynx/Sphynx_Research_Driver.cpp b/packages/zoltan2/test/sphynx/Sphynx_Research_Driver.cpp
index 00d24977885e..0a122582cc61 100644
--- a/packages/zoltan2/test/sphynx/Sphynx_Research_Driver.cpp
+++ b/packages/zoltan2/test/sphynx/Sphynx_Research_Driver.cpp
@@ -409,12 +409,14 @@ int main(int narg, char *arg[])
       std::cout << "init = " << init << std::endl;
     }
 
-    using scalar_type = double;
-    using local_ordinal_type = int;
-    using global_ordinal_type = long long;
+    using scalar_type = Tpetra::Details::DefaultTypes::scalar_type;
+    using local_ordinal_type = Tpetra::Details::DefaultTypes::local_ordinal_type;
+    using global_ordinal_type = Tpetra::Details::DefaultTypes::global_ordinal_type;
     using node_type = Tpetra::Details::DefaultTypes::node_type;
     
     using crs_matrix_type = Tpetra::CrsMatrix<scalar_type, local_ordinal_type, global_ordinal_type, node_type>;  
+    using map_type = Tpetra::Map<local_ordinal_type, global_ordinal_type, node_type>;
+    using mv_type = Tpetra::MultiVector<scalar_type, local_ordinal_type, global_ordinal_type, node_type>;
     using adapter_type = Zoltan2::XpetraCrsGraphAdapter<typename crs_matrix_type::crs_graph_type>;
     using solution_type = Zoltan2::PartitioningSolution<adapter_type>;  
 
@@ -457,14 +459,11 @@ int main(int narg, char *arg[])
       std::cout << "Done with reading/creating the matrix." << std::endl;
     }
 
-    Teuchos::RCP<const Tpetra::Map<> > map = tmatrix->getMap();
+    Teuchos::RCP<const map_type> map = tmatrix->getMap();
 
-    using ST = double;
-    using MultiVector  = Tpetra::MultiVector<ST>;
-
-    Teuchos::RCP<MultiVector > V;
+    Teuchos::RCP<mv_type> V;
     if (vector_file !=""){
-      V = Tpetra::MatrixMarket::Reader<MultiVector >::readDenseFile(vector_file,pComm,map);
+      V = Tpetra::MatrixMarket::Reader<mv_type >::readDenseFile(vector_file,pComm,map);
       if(me == 0){
         std::cout << "Done with reading user-provided eigenvectors." << std::endl;
       }