Merge pull request #59 from pkestene/fix/examples

Fix examples 02 and 03 for CUDA backend
kokkos · Aug 30, 2022 · 23d6aff · 23d6aff
2 parents 003852c + f6de13f
commit 23d6aff
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 5 deletions.
diff --git a/examples/02-axpy-dualview/axpy-dualview-cxx.cc b/examples/02-axpy-dualview/axpy-dualview-cxx.cc
@@ -53,11 +53,12 @@ extern "C" {
     y.template sync<typename view_type::execution_space>();
     x.template sync<typename view_type::execution_space>();
 
+    double d_alpha = *alpha;
     Kokkos::parallel_for( "axpy", y.extent(0), KOKKOS_LAMBDA( const size_t idx)
     {
-      y.d_view(idx) += *alpha * x.d_view(idx);
+      y.d_view(idx) += d_alpha * x.d_view(idx);
     });
-  
+
     y.template modify<typename view_type::execution_space>();
     x.template modify<typename view_type::execution_space>();
     y.template sync<typename view_type::host_mirror_space>();
@@ -67,4 +68,4 @@ extern "C" {
   }
 
 
-}
+}
diff --git a/examples/03-axpy-view/axpy-view-cxx.cc b/examples/03-axpy-view/axpy-view-cxx.cc
@@ -48,13 +48,17 @@ extern "C" {
     view_type y = **v_y;
     view_type x = **v_x;
 
+    double d_alpha = *alpha;
     Kokkos::parallel_for( "axpy", y.extent(0), KOKKOS_LAMBDA( const size_t idx)
     {
-      y(idx) += *alpha * x(idx);
+      y(idx) += d_alpha * x(idx);
     });
 
+    // make sure data can be reused on host
+    Kokkos::fence();
+
     return;
   }
 
 
-}
+}