�
|�e� � �` � d Z ddlmZmZmZ ddlmZmZmZm Z m
Z
m Z m Z m
Z
dgZ G d� de� Zy)z�
pygments.lexers.ptx
~~~~~~~~~~~~~~~~~~~
Lexer for other PTX language.
:copyright: Copyright 2006-2023 by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
� )�
RegexLexer�include�words)�Comment�Keyword�Name�String�Number� Punctuation�
Whitespace�Operator�PtxLexerc
� � e Zd ZdZdZdZdgZdgZdgZdZ dZ
d e z d
z Z d e z d z Z e
d
� e dz ej f e
d� de z ej fdej j" fde z efe ej fdefdefdefdefdefg defdefg ed� ef ed� ej4 f ed � ej6 f ed � ej8 fgd �Z y ) r z}
For NVIDIA `PTX <https://docs.nvidia.com/cuda/parallel-thread-execution/>`_
source.
.. versionadded:: 2.16
�PTXz7https://docs.nvidia.com/cuda/parallel-thread-execution/z*.ptx�ptxz
text/x-ptxz"[^"]*?"z
[a-zA-Z0-9_$]z([-a-zA-Z$._][\w\-$.]*|�)�(z|(\d+))�
whitespacez\s*:�keyword�%z%\d+zc?�;z[*+-/]z0[xX][a-fA-F0-9]+z,-?\d+(?:[.]\d+)?(?:[eE][-+]?\d+(?:[.]\d+)?)?z[=<>{}\[\]()*.,!]|x\bz (\n|\s+)+z//.*?\n)��abs�discard�min�shf�vadd�
activemask�div�mma�shfl�vadd2�add�dp2a�mov�shl�vadd4�addc�dp4a� movmatrix�shr�vavrg2�alloca�elect�mul�sin�vavrg4�and�ex2�mul24�slct�vmad�
applypriority�exit�multimem�sqrt�vmax�atom�fence� nanosleep�st�vmax2�bar�fma�neg� stackrestore�vmax4�barrier�fns�not� stacksave�vmin�bfe�
getctarank�or�stmatrix�vmin2�bfi�griddepcontrol�pmevent�sub�vmin4�bfind�isspacep�popc�subc�vote�bmsk�istypep�prefetch�suld�vset�bra�ld� prefetchu�suq�vset2�brev�ldmatrix�prmt�sured�vset4�brkpt�ldu�rcp�sust�vshl�brx�lg2�red�szext�vshr�call�lop3�redux�tanh�vsub�clz�mad�rem�testp�vsub2�cnot�mad24�ret�tex�vsub4�copysign�madc�rsqrt�tld4�wgmma�cos�mapa�sad�trap�wmma�cp�match�selp�txq�xor� createpolicy�max�set�vabsdiff�cvt�mbarrier�
setmaxnreg� vabsdiff2�cvta�membar�setp� vabsdiff4)
�reg�.sreg�.const�.global�.local�.param�.shared�.texz.wide�.loc)#z
.address_sizez.explicitclusterz.maxnregz.sectionz.aliasz.externz.maxntidr� z.alignz.filez
.minnctapersmr� z.branchtargetsz.funcz .noreturnz.targetz.callprototyper� r� r� z .calltargetsr� z.pragmaz.versionz.commonr� z.regz.visibler� z.maxclusterrankz.reqnctaperclusterz.weakz.entryz
.maxnctapersmz.reqntid)z.s8z.s16z.s32z.s64z.u8z.u16z.u32z.u64z.f16z.f16x2z.f32z.f64z.b8z.b16z.b32z.b64z.pred)�rootr r N) �__name__�
__module__� __qualname__�__doc__�name�url� filenames�aliases� mimetypes�string� followsym�
identifier� block_labelr r �Label�Variable� Anonymousr r r
r
r r r r �Pseudo�Reserved�Type�tokens� � �5/usr/lib/python3/dist-packages/pygments/lexers/ptx.pyr r sv � �� �D�
C�C�� �I��g�G� ��I� �F� �I�+�f�4�s�:�J���#�j�0�K�
�L� !�
�7�
"�D�J�J� /� �I� �
�J�
��
�
� .�
�d�m�m�-�-� .�
�V�^�V� $�
�� � � '�
�;� �
�� !�
!�6� *�
<�f� E�
%�{� 3�#
�* �:� &�
�� !�
� � /� 0�6 29�7
:�: � � � ���
� � 7� 8� :A�9I�9I�
K� � � � � � �
�a7
�5S�Fr� N)r� �pygments.lexerr r r �pygments.tokenr r r r r
r r r
�__all__r r� r� r� �<module>r� s8 ��� 6� 5�&� &� &� �,��f�z� fr� |