tvm_ffi.use_raw_stream#
- tvm_ffi.use_raw_stream(device, stream)[source]#
Create a ffi stream context with given device and stream handle.
- Parameters:
- Return type:
- Returns:
context – The ffi stream context.
Note
When working with torch stram or cuda graph, using
tvm_ffi.use_torch_stream()instead.