MUDA
Loading...
Searching...
No Matches
muda::details::ComputeGraphAccessor Class Reference

Public Member Functions

 ComputeGraphAccessor (ComputeGraph &graph)
 
 ComputeGraphAccessor (ComputeGraph *graph)
 
template<typename T >
void set_kernel_node (const S< KernelNodeParms< T > > &kernelParms)
 
void set_memcpy_node (void *dst, const void *src, size_t size_bytes, cudaMemcpyKind kind)
 
void set_memcpy_node (const cudaMemcpy3DParms &parms)
 
void set_memset_node (const cudaMemsetParams &parms)
 
void set_event_record_node (cudaEvent_t event)
 
void set_event_wait_node (cudaEvent_t event)
 
void set_capture_node (cudaGraph_t sub_graph)
 
auto current_closure () const -> const std::pair< std::string, ComputeGraphClosure * > &
 
auto current_closure () -> std::pair< std::string, ComputeGraphClosure * > &
 
template<typename T >
T * current_node ()
 
const ComputeGraphNodeBasecurrent_node () const
 
ComputeGraphNodeBasecurrent_node ()
 
cudaStream_t current_stream () const
 
cudaStream_t capture_stream () const
 
bool is_topo_built () const
 
void check_allow_var_eval () const
 
void check_allow_node_adding () const
 
template<typename T >
MUDA_INLINE void set_kernel_node (const S< KernelNodeParms< T > > &kernelParms)
 
template<typename T >
MUDA_INLINE void add_kernel_node (const S< KernelNodeParms< T > > &parms)
 
template<typename T >
MUDA_INLINE void update_kernel_node (const S< KernelNodeParms< T > > &kernelParms)
 
template<typename NodeType , typename F >
MUDA_INLINE NodeType * get_or_create_node (F &&f)
 

Friends

class ComputeGraph
 
class muda::ComputeGraphVarBase
 

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