Function tvm::function::ffi::TVMSetStream [−][src]
pub unsafe extern "C" fn TVMSetStream(
device_type: i32,
device_id: i32,
handle: *mut c_void
) -> i32
Expand description
\brief Set the runtime stream of current thread to be stream. The subsequent calls to the same device_type will use the setted stream handle. The specific type of stream is runtime device dependent.
\param device_type The device type. \param device_id The device id. \param handle The stream handle. \return 0 when success, nonzero when failure happens