!ti ` 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-present by the Pygments team, see AUTHORS.
:license: BSD, see LICENSE for details.
)
RegexLexerincludewords)CommentKeywordNameStringNumber Punctuation
WhitespaceOperatorPtxLexerc
e Zd ZdZdZdZdgZdgZdgZdZ dZ
d Z d
e
z d z Z d e z d
z Z
ed e
dz ej f ed de z ej" fdej" j$ fde
z efe ej" fdefdefdefdefdefg defdefg ed ef ed ej6 f ed ej8 f ed ej: fgd Z y )!r za
For NVIDIA `PTX `_
source.
PTXz7https://docs.nvidia.com/cuda/parallel-thread-execution/z*.ptxptxz
text/x-ptxz2.16z"[^"]*?"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)absdiscardminshfvadd
activemaskdivmmashflvadd2adddp2amovshlvadd4addcdp4a movmatrixshrvavrg2allocaelectmulsinvavrg4andex2mul24slctvmad
applypriorityexitmultimemsqrtvmaxatomfence nanosleepstvmax2barfmaneg stackrestorevmax4barrierfnsnot stacksavevminbfe
getctarankorstmatrixvmin2bfigriddepcontrolpmeventsubvmin4bfindisspaceppopcsubcvotebmskistypepprefetchsuldvsetbrald prefetchusuqvset2brevldmatrixprmtsuredvset4brkptldurcpsustvshlbrxlg2redszextvshrcalllop3reduxtanhvsubclzmadremtestpvsub2cnotmad24rettexvsub4copysignmadcrsqrttld4wgmmacosmapasadtrapwmmacpmatchselptxqxor createpolicymaxsetvabsdiffcvtmbarrier
setmaxnreg vabsdiff2cvtamembarsetp 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__nameurl filenamesaliases mimetypes
version_addedstring followsym
identifier block_labelr r LabelVariable Anonymousr r r
r
r r r r PseudoReservedTypetokens b/Volumes/fast/ai/experiments/MLX_z-image/.venv/lib/python3.12/site-packages/pygments/lexers/ptx.pyr r s{ D
CC IgG IM F I+f4s:J#j0K
L !
7
"DJJ / I
J
.
dmm-- .
V^V $
'
;
!
!6 *
r s8 6 5& & & ,ez er