cuda.bindings._v2.driver.stream_update_capture_dependencies_v2#

cuda.bindings._v2.driver.stream_update_capture_dependencies_v2( intptr_t h_stream, dependencies, dependency_data, size_t num_dependencies, unsigned int flags, )#

Update the set of dependencies in a capturing stream.

Modifies the dependency set of a capturing stream. The dependency set is the set of nodes that the next captured node in the stream will depend on along with the edge data for those dependencies. Valid flags are CU_STREAM_ADD_CAPTURE_DEPENDENCIES and CU_STREAM_SET_CAPTURE_DEPENDENCIES. These control whether the set passed to the API is added to the existing set or replaces it. A flags value of 0 defaults to CU_STREAM_ADD_CAPTURE_DEPENDENCIES. Nodes that are removed from the dependency set via this API do not result in CUDA_ERROR_STREAM_CAPTURE_UNJOINED if they are unreachable from the stream at cuStreamEndCapture. Returns CUDA_ERROR_ILLEGAL_STATE if the stream is not capturing.

Parameters:

h_stream (intptr_t) – The stream to update.
dependencies (object) –
The set of dependencies to add. It can be:
- an int as the pointer address to the array, or
- a Python sequence of ints (as pointer addresses).
dependency_data (intptr_t) – Optional array of data associated with each dependency.
num_dependencies (size_t) – The size of the dependencies array.
flags (unsigned int) – See above.