�
2��gL3 � �h � d dl mZ ddlmZ ddlmZ ddlmZ d� ZdCd
�ZdCd�Z ej
ed� � � � � Zej
e ej d
� � d� � � � � � � Z
ej
e ej d� � dDd�� � � � � � Zej
ed� � � � � Zed� � � Zed� � � Zed� � � Zed� � � Zed� � � Zed� � � Zed� � � Zej
e ej ddd�� � dEd �� � � � � � Zej
e ej d!d"�#� � dFd$�� � � � � � Zed%� � � Zed&� � � Zed'� � � Zed(� � � Zej
e ej d)dd�� � dEd*�� � � � � � Zej
e ej d+d"�#� � dFd,�� � � � � � Zed-� � � Z ej
e ej d.� � dGd/�� � � � � � Z!ed0� � � Z"ej
ej# ej d1� � dHd2�� � � � � � Z$ej
e ej% d3� � dId4�� � � � � � Z&ed5� � � Z'ej
e ej% d6� � dId7�� � � � � � Z(edJd9�� � Z)edKd<�� � Z*ej
edej+ fdLd?�� � � � Z,d@� Z-ej
edMdA�� � � � Z.edB� � � Z/dS )N� )�annotations� )�jit� )�core)�mathc �H � t | t j � � r| j n| S �N)�
isinstancer � constexpr�value)�os �h/home/asafur/pinokio/api/open-webui.git/app/env/lib/python3.11/site-packages/triton/language/standard.py�_unwrap_if_constexprr
s � � ��D�N�3�3�:�1�7�7��:� �i�core.constexprc �h � d}| j }|dk r|dz }|dz
}|dk �t j |� � S )Nr r �r
r r )r �log2�ns r �_log2r sF � ��D� ��A�
�a�%�%� �a����� �� �a�%�%� �>�$���r c �X � | j }t j ||dz
z dk o|dk � � S )Nr r r )r r s r �_is_power_of_twor s/ � � ��A��>�1��A��;�1�,�7��a��8�8�8r c � � | |z dz
|z S )z�
Computes the ceiling division of :code:`x` by :code:`div`
:param x: the input number
:type x: Block
:param div: the divisor
:param div: Block
r � )�x�divs r �cdivr ! s � �
��G�a�K�C��r �sigmoidc �8 � ddt j | � � z z S )Nr )r �exp�r s r r r / s � �
��D�H�a�R�L�L� �!�!r �softmaxFc � � | t | d� � z
}t j |� � }t |d� � }t j |||� � S )Nr )�maxr r"