^jdZddlmZddlZddlmZmZmZgdZejddZ ejd dZ y) z&Stream utilities for Inductor codegen.) annotationsN)DEFAULT_STREAMDEFAULT_STREAM_IDXSTREAM_NAME_TEMPLATE)rrrget_raw_stream_nameget_stream_namecD|dk(rtStj|S)zGenerate CUDA Stream name from stream index number. Args: stream_idx: Non-negative index number. 0 refers to the default stream, others refer to side streams. r stream_idx)rrformatr s g/var/www/ramen.bs-engineer-server.com/venv/lib/python3.12/site-packages/torch/_inductor/stream_utils.pyrrs"Q#**jAAc d|S)zCGenerate variable name for a raw stream handle on the given device. raw_stream) device_idxs r rr%s | $$r)r intreturnstr)rrrr) __doc__ __future__r functools torch._inductor.stream_constantsrrr__all__ lru_cacherrrrr rsU,"   B B %%r