Pi ddlmZddlmZddlmcmZddlm Z m Z ddl m Z ddl mZed Zd eed e efd ZeejZeejZeejZeejZeejZeejZe dd Ze ddZe ddZdS))TypeVar) JITFunctionN)GluonJITFunctionjit)knobs)_coreTfnreturnctjjst|tsJt |j}|j|_|SN)rruntime interpret isinstancerrr __doc__)r gluon_fns /var/www/development/aibuddy-work/election-extract/venv/lib/python3.11/site-packages/triton/experimental/gluon/language/_standard.py_import_from_tritonr s> = "Aj[&A&AAAA25zzHzH Oc0tj|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 9UAuf - --rcptj||jn||||jn|| |jjn|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$+sF 9} % } %#^    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% G G GGrr)NNN)typingrtriton.runtime.jitrtriton.language.standardlanguagestandard tl_standard_runtimerrtritonrr rr rcdivsummaxmin reduce_orxor_sumrr$r&rrr7s******.........,,,,,,,, GCLLKN/?/B;+,,+/**+/**+/**   5 6 6  k1 2 2 . . . ., H H H H H Hr