graph: allow access to native graph object

The native graph objects should be accessible for advanced users.
uliegecsm · Mar 28, 2024 · c4a50e1 · c4a50e1
1 parent a53d30a
commit c4a50e1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 0 deletions.
diff --git a/core/src/Cuda/Kokkos_Cuda_Graph_Impl.hpp b/core/src/Cuda/Kokkos_Cuda_Graph_Impl.hpp
@@ -200,6 +200,9 @@ struct GraphImpl<Kokkos::Cuda> {
         m_execution_space, _graph_node_kernel_ctor_tag{},
         aggregate_kernel_impl_t{});
   }
+
+  const cudaGraph_t get_native_graph() const { return m_graph; }
+  const cudaGraphExec_t get_native_executable_graph() const { return m_graph_exec; }
 };
 
 }  // end namespace Impl

diff --git a/core/src/HIP/Kokkos_HIP_Graph_Impl.hpp b/core/src/HIP/Kokkos_HIP_Graph_Impl.hpp
@@ -69,6 +69,9 @@ class GraphImpl<Kokkos::HIP> {
   template <class... PredecessorRefs>
   auto create_aggregate_ptr(PredecessorRefs&&...);
 
+  const hipGraph_t get_native_graph() const { return m_graph; }
+  const hipGraphExec_t get_native_executable_graph() const { return m_graph_exec; }
+
  private:
   void instantiate_graph() {
     constexpr size_t error_log_size = 256;