Ë
    -¼ˆiÍ
  ã                   ó´  — d dl mZ d dlmZ d dlmc mZ ddlm	Z	m
Z
 d dlmZ ddlmZ  ed	«      Zd
ee   de	e   fd„Z eej$                  «      Z eej&                  «      Z eej(                  «      Z eej*                  «      Z eej,                  «      Z eej.                  «      Z eej0                  «      Ze
dd„«       Ze
dd„«       Ze
dd„«       Zy)é    )ÚTypeVar)ÚJITFunctionNé   )ÚGluonJITFunctionÚjit)Úknobsé   )Ú_coreÚTÚfnÚreturnc                 óª   — t         j                  j                  st        | t        «      sJ ‚t        | j                  «      }| j                  |_        |S ©N)r   ÚruntimeÚ	interpretÚ
isinstancer   r   r   Ú__doc__)r   Úgluon_fns     úv/home/obispo/Crisostomo_bridge/mision_env/lib/python3.12/site-packages/triton/experimental/gluon/language/_standard.pyÚ_import_from_tritonr      s<   € Ü=‰=×"Ò"¤j°´[Ô&AÐAÐAä2—5‘5‹z€HØ—z‘z€HÔØ€Oó    c                 ó2   — t        j                  | 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   )ÚttglÚfull©ÚshapeÚdtypeÚlayouts      r   Úzerosr      s   € ô 9‰9U˜A˜u fÓ-Ð-r   c                 ó¢   — t        j                  |€| j                  n|||€| j                  n||€| j                  j
                  «      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.
    )r   r   r   r   Útyper   )ÚinputÚvaluer   r   r   s        r   Ú	full_liker$   ,   sR   € ô 9‰9Ø}ˆŠ¨%ØØ}ˆŠ¨%Ø#˜^ˆ
‰
×Ñó	ð ð 28ó	ð r   c                 ó"   — 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.
    r   r   )r$   )r"   r   r   r   s       r   Ú
zeros_liker&   C   s   € ô U˜A U°%ÀÔGÐGr   r   )NNN)Útypingr   Útriton.runtime.jitr   Útriton.language.standardÚlanguageÚstandardÚtl_standardÚ_runtimer   r   Útritonr   Ú r
   r   r   r   ÚcdivÚsumÚmaxÚminÚravelÚ	reduce_orÚxor_sumr   r$   r&   © r   r   ú<module>r8      sê   ðÝ Ý *ß .Ð .ß ,Ý Ý áˆCƒL€ð˜K¨™Nð Ð/?ÀÑ/Bó ñ ˜;×+Ñ+Ó,€Ù˜+Ÿ/™/Ó*€Ù˜+Ÿ/™/Ó*€Ù˜+Ÿ/™/Ó*€Ù˜K×-Ñ-Ó.€Ù × 5Ñ 5Ó6€	Ù
˜k×1Ñ1Ó
2€ð ò.ó ð.ð òó ðð, òHó ñHr   