tvm_ffi.get_raw_stream

Contents

tvm_ffi.get_raw_stream#

tvm_ffi.get_raw_stream(device)[source]#

Get the current FFI stream of a given device.

Parameters:

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

Return type:

int

Returns:

stream – The current FFI stream as an integer handle.

Examples

import tvm_ffi

dev = tvm_ffi.device("cpu:0")
# Default stream is implementation-defined; set it explicitly
with tvm_ffi.use_raw_stream(dev, 0):
    assert tvm_ffi.get_raw_stream(dev) == 0

See also

tvm_ffi.use_raw_stream()

Set the current stream for a device.

Contents