|
S< ComputeGraph > | create_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 |
|
|
class | ComputeGraph |
|
class | ComputeGraphNodeBase |
|
class | ComputeGraphClosure |
|
The documentation for this class was generated from the following files: