(Th SSKrSSKrSSKrSSKrSSKJr SSKJr SSK J r SSK r SSK r SSK Jr Sr\S:XGa\"\S9r\R%S S S 9 \R%S S \SSSS9 \R%SS\SSS9 \R%SS\SSS9 \R%SS\SSS9 \R%SS\SS S9 \R%S!S"\S#SS$9 \R%S%S&\S'SS$9 \R+5r\R.(a \R.O \R0r\R2(a \R2O\"\5r\"\R45r\R4R9S\"\R:55 \R<R?\R@\5r!\R<RE\!5r#\!RHRK\#5 \&"\#\R05r'\RPRSS(5r(\*"\(5S:Xde\+"\,"S)\RZRSS(555r-S*\ \4S+jr.S,\R^S-\R`3r1\."\-\1/-5r2S.r3\4"\-5VVs0sH)upS/U;dM U4\3"URSS/5S5_M+ snnr5\5Rm5VVs0sH up#UcM X#_M snnr5\4"\-5VVs0sHup\'RnU\3"U5_M snnr8\8Rm5VVs0sH up#UcM X#_M snnr8\5Rm5H#ur9r:\:S:XdM\:\8\'Rn\9S'M% \4"\-5VVs0sH'up\'RnUURSS/5S_M) snnr-\8Hr9S0\-\9'M S1Rw\8Ry5Vs/sH n\"U5PM sn5r=\8Rm5VVs/sH up#US2U3PM snnr>\>S3\R^3S4\R`3/- r>\5Ry5Hr?\?S5;aM S6\?35e \5Rm5VVs0sHup#US7:XdM US8S7//_M snnr@\ RR\'\8\-\@S99rC\R^\R`S:.rD\ R"\C\DS;9rF\FRRS:a\I"S<5e/r7/rJ/rK/rL\4"\'Rn5HurMrN\N\8;aL\7R\N5 \JR\-\N5 \KR\N5 \LR\-\N5 MX\5R\M4S5S:XdMq\7R\N5 \JRS=5 M SrQ\4"\-Ry55HKurMrR\Q\"\M5- rQ\5R\M4S5S:Xa\QS>- rQ\5R\M4S5S7:XdMF\QS?- rQMM S@Rw\\2\Q/5rS\FRSArT\"\R"\T55SBSCrV0SD\S_SE\R0_SF\*"\T5_SGSHRw\W"\VSSSB2\VSSSB25VVs/sH upESIUU3PM snn5_S*SHRw\W"\K\L5VVs/sHupg\"U5SJU3PM snn5_SKSHRw\W"\7\J5VVs/sHupg\"U5SJU3PM snn5_SLSHRw\KVs/sHnSMU3PM snSN/-5_SO\*"\K5S-_SP\>_SQ\FRR_SR\R^_SSS@Rw\=\1/5_ST\(S_SU\(S_SV\(SB_SWS_rYSXHrZ\"\[5R:SY- SZ- S[\Z3- r\\RS\\2S@\QS\\Z35RS]5r_\_R\"\\5R5R"S^0\YD65 SSS5 M ggs snnfs snnfs snnfs snnfs snnfs snfs snnfs snnfs snnfs snnfs snnfs snf!,(df  M=f)_N)ArgumentParser)Path)List) ty_to_cppa Triton ahead-of-time compiler: This program compiles the kernel with name `kernel-name` in the file at the provided `path` into self-contained C source-code that embeds the `cubin` data along with utilities to load, unload and launch the kernel. signature is provided as a list of (optionally divisibility-hinted) types or constexpr values, e.g. `compile.py --kernel-name kernel --signature "*fp32:16, i32:16, 1024, i32" --out-name kernel /path/to/kernel.py` will compile triton.JITFunction of name `kernel` inside the file `/path/to/kernel.py`. Said kernel will be specialized such that argument 0, 1 are assumed to be multiple of 16, and argument 2 is assumed to be a compile-time constant of value 1024, i.e. it won't be part of the generated prototype. The resulting entry point will have signature CUresult kernel_{specialization_suffix}(CUstream stream, unsigned gX, unsigned gY, unsigned gZ, float* arg0, int32_t arg1, int32_t arg2) Different such specialized entry points can be combined using the `linker.py` script. NOTE: when resolving the scope of /path/to/kernel.py, the file will be executed from within its parent directory with the python interpreter used to run this `compile.py` script __main__) descriptionpathzTPath to Python source containing desired kernel in its scope. File will be executed.)helpz --kernel-namez-nzName of the kernel to compileT)typedefaultr requiredz --num-warpsz-wz$Number of warps to launch the kernel)r r r z --num-stagesz-nsz/Number of stages (meta-parameter of the kernel)z --out-namez-onz Out name for the compiled kernelz --out-pathz-oz Out filenamez --signaturez-szSignature of the kernel)r r rz--gridz-gzLaunch grid of the kernel,c$URS5$)N )strip)ss L/var/www/auris/envauris/lib/python3.13/site-packages/triton/tools/compile.pyrFs 1773< signaturec[R"5nURSRU5R 55 UR 5SS$)Nr)hashlibsha256updatejoinencode hexdigest)rms rhash_signaturer#Hs? NN  )$++-.{{}Ra  rwarpsxstagescz[U5nU$![a Of=f[U5nU$![a gf=f)N)int ValueErrorfloat)rrets r constexprr+PsO a&CJ    (CJ   s   - :::r+x=z num_warps=z num_stages=)rz#Only 1 and 16 are valid hints, got r/ztt.divisibility)fn constexprsrattrs) num_warps num_stages)optionszMAOT compiling kernels with global scratch requirements is not yet implementedi32cd_cubin kernel_nametriton_kernel_namebin_sizebin_dataz, 0xrfull_signature arg_pointers&z&global_scratchnum_argskernel_docstringsharedr3 algo_infogridXgridYgridZ _placeholder)hr7extracudazcompile..w)cbinasciirimportlib.util importlibsysargparserpathlibrtypingrtritontriton.backendstriton.backends.nvidia.driverrdesc__name__parser add_argumentstrr' parse_argsargsout_namer=out_pathr arg_pathinsertparentutilspec_from_file_locationstemspecmodule_from_specmodloader exec_modulegetattrkernelgridsplitlenlistmaprr#r3r4meta_sigsig_hashr+ enumeratehintsitems arg_names constantskeyvaluervalues const_sig doc_stringrMr2compiler ASTSourcesrcoptscompileccinfometadataglobal_scratch_size RuntimeError arg_typesarg_names_not_1arg_types_not_1iarg_nameappendgetsuffixty func_nameasmhexlifyhex_ziprGparamsext__file__ template_path with_suffixopenfpwrite read_textformat) rrkvr-ynamerargs 000000000rrs # 34 z -F su CJi!%'  t#qGmn CNP  e#tJlm  dt.Y  t#HDIIHHHOOAs8??+, >> 1 1(-- JD .. ) )$ /CKKC S$** +F 99??3 D t9>>S/1E1Ec1JKLI!$s)! t~~&gdoo->?Hi8*45H I)2):;):QCq*):;JZ/0K?P2QRRJ \\^G|FB1#FF|6;kkm OmdaqBw )Q#R() )m OE // # #v)y`e # fCt GD ^^C .F **Q.jkkIIOO !1!12 8 9 $   X &   Yx0 1  " "8 ,  " "9X#6 7 YYud #q (   X &   U #3F9++-.2#a& 99aUD !Q & cMF 99aUD !R ' cMF / (Hf56I **W C x$ %a +Dyd.. CH DIIs4!9d14a4j7QR7QtqA3qc{7QRS  TYY#o_nJopJohd9R=/4& 9Jopq  $))sS\^gOh$iOh84 " av%>Oh$ij   "HQse9"HL]K^"^_ C(1, J &//(( T^^ SXXy(34 a a a !F$X--7&@XcUCSS  ! !AhZq#"? @ E Ec Jb HHT-(224;;EfE FK Jql ] =TEX>; PHSp$i"HK Jsl3 `' `': `-`-"`3 `9"`9..`? a7a ) a9 aa6a/a" a( #6a-- a<