o Ù°“i¯ã@s¬dZddlmZddlmZddlmZddlmZddl m Z Gdd „d ƒZ zdd l Z Gd d „d ƒZ dddd„ZWneyIdddd„ZYnwddd„Zddd„Zd S) zStream context.é)Ú annotations)Úc_void_p)ÚAnyé)Úcore©Údevicec@s.eZdZdZddd „Zdd d „Zddd„ZdS)Ú StreamContexta1Represent a stream context in the FFI system. StreamContext helps setup ffi environment stream by python `with` statement. When entering `with` scope, it caches the current environment stream and setup the given new stream. When exiting `with` scope, it recovers the stream to the cached environment stream. Parameters ---------- device The device to which the stream belongs. stream The stream handle. See Also -------- :py:func:`tvm_ffi.use_raw_stream`, :py:func:`tvm_ffi.use_torch_stream` rú core.DeviceÚstreamúint | c_void_pÚreturnÚNonecCs| ¡|_|j|_||_dS)zs      * #þ .