
    3j                    j    S r SSKJr  SSKrSSKJrJrJr  / SQr\R                  SS j5       r
S	S jrg)
z&Stream utilities for Inductor codegen.    )annotationsN)DEFAULT_STREAMDEFAULT_STREAM_IDXSTREAM_NAME_TEMPLATE)r   r   r   get_raw_stream_nameget_stream_namec                B    U S:X  a  [         $ [        R                  " U S9$ )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)r   r   formatr
   s    V/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/_inductor/stream_utils.pyr   r      s"     Q#**jAA    c                    SU  3$ )zCGenerate variable name for a raw stream handle on the given device.
raw_stream )
device_idxs    r   r   r   %   s    
|$$r   )r   intreturnstr)r   r   r   r   )__doc__
__future__r   	functools torch._inductor.stream_constantsr   r   r   __all__	lru_cacher   r   r   r   r   <module>r      s@    , "   
B 
B%r   