99"""
1010
1111from pygments .lexer import RegexLexer , include , words
12- from pygments .token import Comment , Keyword , Name , String , Number , Punctuation , \
13- Whitespace , Operator
12+ from pygments .token import Comment , Keyword , Name , String , Number , \
13+ Punctuation , Whitespace , Operator
1414
1515__all__ = ["PtxLexer" ]
1616
@@ -20,7 +20,7 @@ class PtxLexer(RegexLexer):
2020 For NVIDIA `PTX <https://docs.nvidia.com/cuda/parallel-thread-execution/>`_
2121 source.
2222
23- .. versionadded:: 2.15
23+ .. versionadded:: 2.16
2424 """
2525 name = 'PTX'
2626 url = "https://docs.nvidia.com/cuda/parallel-thread-execution/"
@@ -65,31 +65,31 @@ class PtxLexer(RegexLexer):
6565 (words ((
6666 'abs' , 'discard' , 'min' , 'shf' , 'vadd' ,
6767 'activemask' , 'div' , 'mma' , 'shfl' , 'vadd2' ,
68- 'add' ,'dp2a' ,'mov' ,'shl' ,'vadd4' ,
69- 'addc' ,'dp4a' ,'movmatrix' ,'shr' ,'vavrg2' ,
70- 'alloca' ,'elect' ,'mul' ,'sin' ,'vavrg4' ,
71- 'and' ,'ex2' ,'mul24' ,'slct' ,'vmad' ,
72- 'applypriority' ,'exit' ,'multimem' ,'sqrt' ,'vmax' ,
73- 'atom' ,'fence' ,'nanosleep' ,'st' ,'vmax2' ,
74- 'bar' ,'fma' ,'neg' ,'stackrestore' ,'vmax4' ,
75- 'barrier' ,'fns' ,'not' ,'stacksave' ,'vmin' ,
76- 'bfe' ,'getctarank' ,'or' ,'stmatrix' ,'vmin2' ,
77- 'bfi' ,'griddepcontrol' ,'pmevent' ,'sub' ,'vmin4' ,
78- 'bfind' ,'isspacep' ,'popc' ,'subc' ,'vote' ,
79- 'bmsk' ,'istypep' ,'prefetch' ,'suld' ,'vset' ,
80- 'bra' ,'ld' , 'prefetchu' ,'suq' ,'vset2' ,
81- 'brev' ,'ldmatrix' ,'prmt' ,'sured' ,'vset4' ,
82- 'brkpt' ,'ldu' ,'rcp' ,'sust' ,'vshl' ,
83- 'brx' ,'lg2' ,'red' ,'szext' ,'vshr' ,
84- 'call' ,'lop3' ,'redux' ,'tanh' ,'vsub' ,
85- 'clz' ,'mad' , 'rem' ,'testp' ,'vsub2' ,
86- 'cnot' ,'mad24' ,'ret' ,'tex' ,'vsub4' ,
87- 'copysign' ,'madc' ,'rsqrt' ,'tld4' ,'wgmma' ,
88- 'cos' ,'mapa' ,'sad' ,'trap' ,'wmma' ,
89- 'cp' ,'match' ,'selp' ,'txq' ,'xor' ,
90- 'createpolicy' ,'max' ,'set' ,'vabsdiff' ,'cvt' ,
91- 'mbarrier' ,'setmaxnreg' ,'vabsdiff2' ,'cvta' ,
92- 'membar' ,'setp' ,'vabsdiff4' )), Keyword ),
68+ 'add' , 'dp2a' , 'mov' , 'shl' , 'vadd4' ,
69+ 'addc' , 'dp4a' , 'movmatrix' , 'shr' , 'vavrg2' ,
70+ 'alloca' , 'elect' , 'mul' , 'sin' , 'vavrg4' ,
71+ 'and' , 'ex2' , 'mul24' , 'slct' , 'vmad' ,
72+ 'applypriority' , 'exit' , 'multimem' , 'sqrt' , 'vmax' ,
73+ 'atom' , 'fence' , 'nanosleep' , 'st' , 'vmax2' ,
74+ 'bar' , 'fma' , 'neg' , 'stackrestore' , 'vmax4' ,
75+ 'barrier' , 'fns' , 'not' , 'stacksave' , 'vmin' ,
76+ 'bfe' , 'getctarank' , 'or' , 'stmatrix' , 'vmin2' ,
77+ 'bfi' , 'griddepcontrol' , 'pmevent' , 'sub' , 'vmin4' ,
78+ 'bfind' , 'isspacep' , 'popc' , 'subc' , 'vote' ,
79+ 'bmsk' , 'istypep' , 'prefetch' , 'suld' , 'vset' ,
80+ 'bra' , 'ld' , 'prefetchu' , 'suq' , 'vset2' ,
81+ 'brev' , 'ldmatrix' , 'prmt' , 'sured' , 'vset4' ,
82+ 'brkpt' , 'ldu' , 'rcp' , 'sust' , 'vshl' ,
83+ 'brx' , 'lg2' , 'red' , 'szext' , 'vshr' ,
84+ 'call' , 'lop3' , 'redux' , 'tanh' , 'vsub' ,
85+ 'clz' , 'mad' , 'rem' , 'testp' , 'vsub2' ,
86+ 'cnot' , 'mad24' , 'ret' , 'tex' , 'vsub4' ,
87+ 'copysign' , 'madc' , 'rsqrt' , 'tld4' , 'wgmma' ,
88+ 'cos' , 'mapa' , 'sad' , 'trap' , 'wmma' ,
89+ 'cp' , 'match' , 'selp' , 'txq' , 'xor' ,
90+ 'createpolicy' , 'max' , 'set' , 'vabsdiff' , 'cvt' ,
91+ 'mbarrier' , 'setmaxnreg' , 'vabsdiff2' , 'cvta' ,
92+ 'membar' , 'setp' , 'vabsdiff4' )), Keyword ),
9393 # State Spaces and Suffixes
9494 (words ((
9595 'reg' , '.sreg' , '.const' , '.global' ,
@@ -98,15 +98,15 @@ class PtxLexer(RegexLexer):
9898 )), Keyword .Pseudo ),
9999 # PTX Directives
100100 (words ((
101- '.address_size' ,'.explicitcluster' ,'.maxnreg' ,'.section' ,
102- '.alias' , '.extern' ,'.maxntid' ,'.shared' ,
103- '.align' ,'.file' ,'.minnctapersm' ,'.sreg' ,
104- '.branchtargets' ,'.func' ,'.noreturn' ,'.target' ,
105- '.callprototype' ,'.global' ,'.param' ,'.tex' ,
106- '.calltargets' ,'.loc' ,'.pragma' ,'.version' ,
107- '.common' ,'.local' ,'.reg' ,'.visible' ,
108- '.const' ,'.maxclusterrank' ,'.reqnctapercluster' ,'.weak' ,
109- '.entry' ,'.maxnctapersm' ,'.reqntid' )), Keyword .Reserved ),
101+ '.address_size' , '.explicitcluster' , '.maxnreg' , '.section' ,
102+ '.alias' , '.extern' , '.maxntid' , '.shared' ,
103+ '.align' , '.file' , '.minnctapersm' , '.sreg' ,
104+ '.branchtargets' , '.func' , '.noreturn' , '.target' ,
105+ '.callprototype' , '.global' , '.param' , '.tex' ,
106+ '.calltargets' , '.loc' , '.pragma' , '.version' ,
107+ '.common' , '.local' , '.reg' , '.visible' ,
108+ '.const' , '.maxclusterrank' , '.reqnctapercluster' , '.weak' ,
109+ '.entry' , '.maxnctapersm' , '.reqntid' )), Keyword .Reserved ),
110110 # Fundamental Types
111111 (words ((
112112 '.s8' , '.s16' , '.s32' , '.s64' ,
0 commit comments