TrilinosCouplings: Initialize Kokkos in Maxwell Epetra example

MueLu RefMaxwell requires Kokkos
trilinos · Sep 1, 2023 · 2bb4069 · 2bb4069
1 parent cb7a065
commit 2bb4069
Show file tree

Hide file tree

Showing 2 changed files with 33 additions and 16 deletions.
diff --git a/packages/trilinoscouplings/examples/scaling/CMakeLists.txt b/packages/trilinoscouplings/examples/scaling/CMakeLists.txt
@@ -54,29 +54,34 @@ IF(${PACKAGE_NAME}_ENABLE_Epetra AND ${PACKAGE_NAME}_ENABLE_EpetraExt AND
     NUM_MPI_PROCS 1
     )
 
-  TRIBITS_ADD_EXECUTABLE(
-    Example_Maxwell
-    SOURCES example_Maxwell.cpp
-      TrilinosCouplings_Pamgen_Utils.cpp
-    COMM serial mpi
+
+  IF(${PACKAGE_NAME}_ENABLE_Kokkos)
+
+    TRIBITS_ADD_EXECUTABLE(
+      Example_Maxwell
+      SOURCES example_Maxwell.cpp
+        TrilinosCouplings_Pamgen_Utils.cpp
+      COMM serial mpi
     )
 
-  TRIBITS_ADD_TEST(
+    TRIBITS_ADD_TEST(
       Example_Maxwell
       NAME Example_Maxwell_ML
       ARGS "--solverName=ML"
       NUM_MPI_PROCS 1
       COMM mpi serial
-      )
+    )
 
-  IF(${PACKAGE_NAME}_ENABLE_Stratimikos)
-    TRIBITS_ADD_TEST(
-      Example_Maxwell
-      NAME Example_Maxwell_ML_Stratimikos
-      ARGS "--solverName=ML-Stratimikos"
-      NUM_MPI_PROCS 1
-      COMM mpi serial
+    IF(${PACKAGE_NAME}_ENABLE_Stratimikos)
+      TRIBITS_ADD_TEST(
+        Example_Maxwell
+        NAME Example_Maxwell_ML_Stratimikos
+        ARGS "--solverName=ML-Stratimikos"
+        NUM_MPI_PROCS 1
+        COMM mpi serial
       )
+    ENDIF()
+
   ENDIF()
 
 
@@ -95,7 +100,7 @@ IF(${PACKAGE_NAME}_ENABLE_Epetra AND ${PACKAGE_NAME}_ENABLE_EpetraExt AND
       Example_Poisson2D_pn_tpetra
       SOURCES example_Poisson2D_pn_tpetra.cpp
       TrilinosCouplings_Pamgen_Utils.cpp
-      TrilinosCouplings_IntrepidPoissonExampleHelpers.cpp	
+      TrilinosCouplings_IntrepidPoissonExampleHelpers.cpp
       COMM mpi
       )
 
@@ -107,7 +112,7 @@ IF(${PACKAGE_NAME}_ENABLE_Epetra AND ${PACKAGE_NAME}_ENABLE_EpetraExt AND
       NUM_MPI_PROCS 1
       )
 
-    IF(MueLu_ENABLE_Epetra)
+    IF(MueLu_ENABLE_Epetra AND ${PACKAGE_NAME}_ENABLE_Kokkos)
       TRIBITS_ADD_TEST(
         Example_Maxwell
         NAME Example_Maxwell_MueLu

diff --git a/packages/trilinoscouplings/examples/scaling/example_Maxwell.cpp b/packages/trilinoscouplings/examples/scaling/example_Maxwell.cpp
@@ -98,6 +98,9 @@
 #include "Intrepid_DefaultCubatureFactory.hpp"
 #include "Intrepid_Utils.hpp"
 
+// Kokkos includes
+#include "Kokkos_Core.hpp"
+
 // Epetra includes
 #include "Epetra_Time.h"
 #include "Epetra_Map.h"
@@ -403,7 +406,16 @@ int evalCurlCurlu(double & curlcurlu0,
 /**********************************************************************************/
 /******************************** MAIN ********************************************/
 /**********************************************************************************/
+int body(int argc, char *argv[]);
+
 int main(int argc, char *argv[]) {
+  Kokkos::initialize(argc,argv);
+  body(argc,argv);
+  if (!Kokkos::is_finalized())
+    Kokkos::finalize();
+}
+
+int body(int argc, char *argv[]) {
   // using namespace TrilinosCouplings;
   // using IntrepidPoissonExample::parseCommandLineArguments;
   // using IntrepidPoissonExample::setCommandLineArgumentDefaults;