cuda.bindings._v2.driver.graph_add_host_node#

cuda.bindings._v2.driver.graph_add_host_node( intptr_t h_graph, dependencies, size_t num_dependencies, node_params, ) → intptr_t#

Creates a host execution node and adds it to a graph.

Creates a new CPU execution node and adds it to h_graph with num_dependencies dependencies specified via dependencies and arguments specified in node_params. It is possible for num_dependencies to be 0, in which case the node will be placed at the root of the graph. dependencies may not have any duplicate entries. A handle to the new node will be returned in ph_graph_node. When the graph is launched, the node will invoke the specified CPU function. Host nodes are not supported under MPS with pre-Volta GPUs.

Parameters:

h_graph (intptr_t) – Graph to which to add the node.
dependencies (object) –
Dependencies of the node. It can be:
- an int as the pointer address to the array, or
- a Python sequence of ints (as pointer addresses).
num_dependencies (size_t) – Number of dependencies.
node_params (intptr_t) – Parameters for the host node.

Returns:

Returns newly created node.

Return type:

intptr_t