Add initialization and finalization in examples

kokkos · Jan 13, 2025 · 19bd963 · 19bd963
1 parent 284a0fb
commit 19bd963
Show file tree

Hide file tree

Showing 10 changed files with 22 additions and 2 deletions.
diff --git a/examples/01_1DFFT/01_1DFFT.cpp b/examples/01_1DFFT/01_1DFFT.cpp
@@ -13,6 +13,7 @@ using View1D = Kokkos::View<T*, execution_space>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -44,6 +45,7 @@ int main(int argc, char* argv[]) {
     KokkosFFT::irfft(exec, xc2r, xc2r_hat);
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/examples/02_2DFFT/02_2DFFT.cpp b/examples/02_2DFFT/02_2DFFT.cpp
@@ -13,6 +13,7 @@ using View2D = Kokkos::View<T**, execution_space>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128, n1 = 128;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -44,6 +45,7 @@ int main(int argc, char* argv[]) {
     KokkosFFT::irfft2(exec, xc2r, xc2r_hat);
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/examples/03_NDFFT/03_NDFFT.cpp b/examples/03_NDFFT/03_NDFFT.cpp
@@ -15,6 +15,7 @@ using axis_type = KokkosFFT::axis_type<DIM>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128, n1 = 128, n2 = 16;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -46,6 +47,7 @@ int main(int argc, char* argv[]) {
     KokkosFFT::irfftn(exec, xc2r, xc2r_hat, axis_type<3>{-3, -2, -1});
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/examples/04_batchedFFT/04_batchedFFT.cpp b/examples/04_batchedFFT/04_batchedFFT.cpp
@@ -13,6 +13,7 @@ using View3D = Kokkos::View<T***, execution_space>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128, n1 = 128, n2 = 16;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -48,6 +49,7 @@ int main(int argc, char* argv[]) {
                      /*axis=*/-1);
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/examples/05_1DFFT_HOST_DEVICE/05_1DFFT_HOST_DEVICE.cpp b/examples/05_1DFFT_HOST_DEVICE/05_1DFFT_HOST_DEVICE.cpp
@@ -16,6 +16,7 @@ using HostView1D = Kokkos::View<T*, host_execution_space>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -79,6 +80,7 @@ int main(int argc, char* argv[]) {
     host_exec.fence();
 #endif
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/examples/06_1DFFT_reuse_plans/06_1DFFT_reuse_plans.cpp b/examples/06_1DFFT_reuse_plans/06_1DFFT_reuse_plans.cpp
@@ -13,6 +13,7 @@ using View1D = Kokkos::View<T*, execution_space>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -54,6 +55,7 @@ int main(int argc, char* argv[]) {
     irfft_plan.execute(xc2r, xc2r_hat);
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/examples/07_unmanaged_views/07_unmanaged_views.cpp b/examples/07_unmanaged_views/07_unmanaged_views.cpp
@@ -25,6 +25,7 @@ using shape_type = KokkosFFT::shape_type<DIM>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128, n1 = 128, n2 = 16;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -63,6 +64,7 @@ int main(int argc, char* argv[]) {
                      KokkosFFT::Normalization::backward, shape);
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/examples/08_inplace_FFT/08_inplace_FFT.cpp b/examples/08_inplace_FFT/08_inplace_FFT.cpp
@@ -14,6 +14,7 @@ using RightView2D = Kokkos::View<T **, Kokkos::LayoutRight, execution_space>;
 
 int main(int argc, char *argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
     const int n0 = 128, n1 = 128;
     const Kokkos::complex<double> z(1.0, 1.0);
@@ -69,6 +70,7 @@ int main(int argc, char *argv[]) {
 
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/install_test/as_library/hello.cpp b/install_test/as_library/hello.cpp
@@ -13,8 +13,9 @@ using View1D = Kokkos::View<T*, execution_space>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
-    constexpr int n0 = 128;
+    const int n0 = 128;
     const Kokkos::complex<double> z(1.0, 1.0);
 
     // 1D C2C FFT (Forward and Backward)
@@ -44,6 +45,7 @@ int main(int argc, char* argv[]) {
     KokkosFFT::irfft(exec, xc2r, xc2r_hat);
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;

diff --git a/install_test/as_subdirectory/hello.cpp b/install_test/as_subdirectory/hello.cpp
@@ -13,8 +13,9 @@ using View2D = Kokkos::View<T**, execution_space>;
 
 int main(int argc, char* argv[]) {
   Kokkos::initialize(argc, argv);
+  KokkosFFT::initialize();
   {
-    constexpr int n0 = 128, n1 = 128;
+    const int n0 = 128, n1 = 128;
     const Kokkos::complex<double> z(1.0, 1.0);
 
     // 2D C2C FFT (Forward and Backward)
@@ -44,6 +45,7 @@ int main(int argc, char* argv[]) {
     KokkosFFT::irfft2(exec, xc2r, xc2r_hat);
     exec.fence();
   }
+  KokkosFFT::finalize();
   Kokkos::finalize();
 
   return 0;