|
| 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 ComputeGraphNodeBase * | current_node () const |
|
ComputeGraphNodeBase * | current_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) |
|
|
class | ComputeGraph |
|
class | muda::ComputeGraphVarBase |
|
The documentation for this class was generated from the following files: