cuda.bindings._v2.driver.graph_add_memset_node#

cuda.bindings._v2.driver.graph_add_memset_node( intptr_t h_graph, dependencies, size_t num_dependencies, memset_params, intptr_t ctx, ) → intptr_t#

Creates a memset node and adds it to a graph.

Creates a new memset node and adds it to h_graph with num_dependencies dependencies specified via dependencies. 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. The element size must be 1, 2, or 4 bytes. When the graph is launched, the node will perform the memset described by memset_params.

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.
memset_params (intptr_t) – Parameters for the memory set.
ctx (intptr_t) – Context on which to run the node.

Returns:

Returns newly created node.

Return type:

intptr_t