Fencing for sptrsv

kokkos · Sep 26, 2023 · 8497ad2 · 8497ad2
1 parent 82a2050
commit 8497ad2
Showing 1 changed file with 6 additions and 4 deletions.
diff --git a/sparse/impl/KokkosSparse_twostage_gauss_seidel_impl.hpp b/sparse/impl/KokkosSparse_twostage_gauss_seidel_impl.hpp
@@ -986,10 +986,11 @@ class TwostageGaussSeidel {
                 Kokkos::subview(localZ, Kokkos::ALL(), range_type(j, j + 1));
             single_vector_view_t Rj(localRj.data(), num_rows);
             single_vector_view_t Zj(localZj.data(), num_rows);
+            my_exec_space
+                .fence();  // Wait for writes to R and Z to land, TODO: remove
             sptrsv_solve(handle->get_gs_sptrsvL_handle(), crsmatL.graph.row_map,
                          crsmatL.graph.entries, crsmatL.values, Rj, Zj);
-            execution_space()
-                .fence();  // TODO: call sptrsv_solve on stream and remove
+            Kokkos::fence();  // TODO: call sptrsv_solve on stream and remove
           }
         } else {
           using namespace KokkosSparse::Experimental;
@@ -1002,10 +1003,11 @@ class TwostageGaussSeidel {
                 Kokkos::subview(localZ, Kokkos::ALL(), range_type(j, j + 1));
             single_vector_view_t Rj(localRj.data(), num_rows);
             single_vector_view_t Zj(localZj.data(), num_rows);
+            my_exec_space
+                .fence();  // Wait for writes to R and Z to land, TODO: remove
             sptrsv_solve(handle->get_gs_sptrsvU_handle(), crsmatU.graph.row_map,
                          crsmatU.graph.entries, crsmatU.values, Rj, Zj);
-            execution_space()
-                .fence();  // TODO: call sptrsv_solve on stream and remove
+            Kokkos::fence();  // TODO: call sptrsv_solve on stream and remove
           }
         }