MueLu LWGraphBase: Allow calling GetNodeNumEdges from host and device

Signed-off-by: Christian Glusa <caglusa@sandia.gov>
trilinos · Sep 11, 2024 · e39148c · e39148c
1 parent 8e1e152
commit e39148c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/packages/muelu/src/Graph/Containers/MueLu_LWGraphBase.hpp b/packages/muelu/src/Graph/Containers/MueLu_LWGraphBase.hpp
@@ -187,7 +187,7 @@ class LWGraphBase {
 
   //! Return number of graph edges
   KOKKOS_INLINE_FUNCTION size_type GetNodeNumEdges() const {
-    return graph_.row_map(GetNodeNumVertices());
+    return graph_.entries.extent(0);
   }
 
   //! Return global number of graph edges