Function TVMFFIEnvSetStream

Contents

Function TVMFFIEnvSetStream#

Defined in File c_env_api.h

Function Documentation#

int TVMFFIEnvSetStream(int32_t device_type, int32_t device_id, TVMFFIStreamHandle stream, TVMFFIStreamHandle *opt_out_original_stream)#

FFI function to set the current stream for a device.

Parameters:

device_type – The type of the device.
device_id – The id of the device.
stream – The stream to set.
opt_out_original_stream – Output original stream if the address is not nullptr.

Returns:

0 when success, nonzero when failure happens