muda
Public Member Functions | Friends | List of all members
muda::ComputeGraphVarManager Class Reference

Public Member Functions

S< ComputeGraphcreate_graph (std::string_view name="graph", ComputeGraphFlag flags={})
 
template<typename T >
ComputeGraphVar< T > & create_var (std::string_view name)
 
template<typename T >
ComputeGraphVar< T > & create_var (std::string_view name, const T &init_value)
 
template<typename T >
ComputeGraphVar< T > * find_var (std::string_view name)
 
bool is_using () const
 
void sync () const
 
void sync_on (cudaStream_t stream) const
 
template<typename... T>
bool is_using (const ComputeGraphVar< T > &... vars) const
 
template<typename... T>
void sync (const ComputeGraphVar< T > &... vars) const
 
template<typename... T>
void sync_on (cudaStream_t stream, const ComputeGraphVar< T > &... vars) const
 
bool is_using (const span< const ComputeGraphVarBase *> vars) const
 
void sync (const span< const ComputeGraphVarBase *> vars) const
 
void sync_on (cudaStream_t stream, const span< const ComputeGraphVarBase *> vars) const
 
const auto & graphs () const
 
void graphviz (std::ostream &os, const ComputeGraphGraphvizOptions &options={}) const
 
template<typename T >
MUDA_INLINE ComputeGraphVar< T > & create_var (std::string_view name)
 
template<typename T >
MUDA_INLINE ComputeGraphVar< T > & create_var (std::string_view name, const T &init_value)
 
template<typename T >
MUDA_INLINE ComputeGraphVar< T > * find_var (std::string_view name)
 
template<typename... T>
MUDA_INLINE bool is_using (const ComputeGraphVar< T > &... vars) const
 
template<typename... T>
MUDA_INLINE void sync (const ComputeGraphVar< T > &... vars) const
 
template<typename... T>
MUDA_INLINE void sync_on (cudaStream_t stream, const ComputeGraphVar< T > &... vars) const
 

Friends

class ComputeGraph
 
class ComputeGraphNodeBase
 
class ComputeGraphClosure
 

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