nvmath.bindings.curand.set_stream

nvmath.bindings.curand.set_stream(intptr_t generator, intptr_t stream)[source]

Set the current stream for CURAND kernel launches.

Parameters:

  • generator (intptr_t) – Generator to modify.

  • stream (intptr_t) – Stream to use or NULL for null stream.

See also

curandSetStream