c i ddlmZddlmZddlmcmZddlm Z m Z ddl m Z ddl mZed Zd eed e efd Zeej$Zeej&Zeej(Zeej*Zeej,Zeej.Ze dd Ze ddZe ddZy))TypeVar) JITFunctionN)GluonJITFunctionjit)knobs)_coreTfnreturnctjjst|tsJt |j }|j|_|SN)rruntime interpret isinstancerrr __doc__)r gluon_fns f/var/www/html/engine/venv/lib/python3.12/site-packages/triton/experimental/gluon/language/_standard.py_import_from_tritonr s< == " "j[&AA A255zHzzH Oc2tj|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)ttglfullshapedtypelayouts rzerosrs 99UAuf --rctj| |jn||| |jn|||jj S|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. )rrrrtyper)inputvaluerrrs r full_liker$+sR 99} % } %#^   28  rc"t|d|||S)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$)r"rrrs r zeros_liker&Bs UAU% GGrr)NNN)typingrtriton.runtime.jitrtriton.language.standardlanguagestandard tl_standard_runtimerrtritonrr rr rcdivsummaxmin reduce_orxor_sumrr$r&rrr7s*.., CLKN/?/B;++,+//*+//*+//*  5 5 6 k11 2 . ., H Hr