o i @sddlmZddlmZddlmmZddlm Z m Z ddl m Z ddl mZed Zd eed e efd d ZeejZeejZeejZeejZeejZeejZe dddZe dddZe dddZdS))TypeVar) JITFunctionN)GluonJITFunctionjit)knobs)_coreTfnreturncCs,tjjs t|ts Jt|j}|j|_|SN)rruntime interpret isinstancerrr __doc__)r gluon_fnrb/home/ubuntu/vllm_env/lib/python3.10/site-packages/triton/experimental/gluon/language/_standard.py_import_from_triton s rcCst|d||S)aV Create a tensor filled with zeros. Args: shape (Sequence[int]): The shape of the tensor. dtype (dtype): The data type for the tensor. layout (Optional[DistributedLayout]): The distributed layout of the tensor, defaults to AutoLayout(). Returns: tensor: A tensor where every element is zero. r)ttglfullshapedtypelayoutrrrzeross rcCs>t|dur |jn|||dur|jn||dur|jjS|S)a- Create a tensor with the same properties as a given tensor, filled with a specified value. Args: input (tensor): Reference tensor to infer default shape, dtype, and layout. value (int or float): The fill value. shape (Sequence[int], optional): Target shape. Defaults to input.shape. dtype (dtype, optional): Target data type. Defaults to input.dtype. layout (DistributedLayout, optional): Target layout. Defaults to input.layout. Returns: tensor: A tensor where every element equals value. N)rrrrtyper)inputvaluerrrrrr full_like+sr cCst|d|||dS)a Create a tensor with the same properties as a given tensor, filled with zeros. Args: input (tensor): Reference tensor to infer default shape, dtype, and layout. shape (Sequence[int], optional): Target shape. Defaults to input.shape. dtype (dtype, optional): Target data type. Defaults to input.dtype. layout (DistributedLayout, optional): Target layout. Defaults to input.layout. Returns: tensor: A tensor where every element is zero. rr)r )rrrrrrr zeros_likeBsr!r )NNN)typingrtriton.runtime.jitrtriton.language.standardlanguagestandard tl_standard_runtimerrtritonrr rr rcdivsummaxmin reduce_orxor_sumrr r!rrrrs(