Function TVMFFIEnvGetStream#
Defined in File c_env_api.h
Function Documentation#
-
TVMFFIStreamHandle TVMFFIEnvGetStream(int32_t device_type, int32_t device_id)#
FFI function to get the current stream for a device.
- Parameters:
device_type – The type of the device.
device_id – The id of the device.
- Returns:
The current stream of the device.