Collective Variables Module - Developer Documentation
Loading...
Searching...
No Matches
Public Member Functions | Public Attributes | List of all members
colvars_gpu::gpu_graph_t Class Reference

A struct for holding a CUDA graph and its execution object. More...

#include <colvar_gpu_support.h>

Public Member Functions

 gpu_graph_t ()
 Constructor.
 
int reset ()
 Reset the CUDA graph.
 
 ~gpu_graph_t ()
 Destructor.
 
int dump_graph (const std::string &filename)
 Dump the CUDA graph to a dot file for debugging.
 
int init_graph_exec (colvarmodule *cvmodule, cudaStream_t stream)
 Initialize CUDA graph execution instance.
 

Public Attributes

bool graph_exec_initialized = false
 Flag to describe whether the graph execution instance has been initialized.
 
cudaGraph_t graph = nullptr
 CUDA graph object.
 
cudaGraphExec_t graph_exec = nullptr
 CUDA graph execution instance object.
 
std::unordered_map< std::string, cudaGraphNode_t > nodes = {}
 List of compute nodes.
 

Detailed Description

A struct for holding a CUDA graph and its execution object.


The documentation for this class was generated from the following files: