tvm_ffi.use_raw_stream

tvm_ffi.use_raw_stream(device, stream)

Create a ffi stream context with given device and stream handle.

Parameters:

• device (tvm_ffi.Device) – The device to which the stream belongs.

• stream (Union[int, c_void_p]) – The stream handle.

Returns:

context – The ffi stream context.

Return type:

tvm_ffi.StreamContext

Note

When working with torch stram or cuda graph, using tvm_ffi.use_torch_stream() instead.