+ j3^RIt^RIt^RIt^RIt^RIt^RIHtRRltRRltRRlt R R lt R R lt R RRllt ]!R4R 3RRllt ]R8Xd]P!RR7t]P#R]RR7]P#R]]!R4RR7]P#RRRR7]P%4t] !]P(]P*]P,'*4tR#R#)!N)Pathc0V^8dQhR\R\/# source_codereturnstr)formats"^/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/utils/_get_clean_triton.py __annotate__r sCCcf\P!RRV4p\P!RRV4pV#)z(\n.+\s\'\'\'\n) z (\'\'\'\,.+))resub)r remove_head remove_tails& r "remove_triton_function_declarationr s-&&,dK@K&&$ %?$$rcaRpRV3Rllp\P!W#V4p\P!RRV\PR7pV#)z(\w+)\.run\((.*)\)c$V^8dQhR\/#)rrr )r s"r r 'add_launch_params..__annotate__>s  # rc<VP^4pVP^4pSV,wr4\VPR4VPR44pV RV RRPV4 R2pV#)z, [z]())r$r8splitjoin)r* func_nameparamsr2grid new_stringr;s& r r"add_launch_params..replace>stKKN Q+I6 !&,,t"4j6F6Ft6LM "{!D6DIIj,A+B!D rz"@triton_heuristics[^@]*@triton.jitz @triton.jit)flags)rrr")r:r;r(r transformedremove_inductor_wrapperss&f r add_launch_paramsrO8sM$G  &&84K!vv-ii   $#rTcHV^8dQhR\R\R\R\/#)rinput_filenameoutput_filenameauto_generate_paramsr)r bool)r s"r r r Us,???*-?EI??rc \V4;_uu_4pVP4pRRR4XpRV9d \R4h\V4p\ V4pV R2p\ P PV4'EgV'd\RV R24\ PP4pRVR&\P!\PV.TRR\ P PV4;'gR R 7pVP ^8wdX\R V R 24\R VP" 24\RVP$ 24\RVP 24h\RV 24\ P PV4'g\RV RV R24h\V4;_uu_4p V P+4p RRR4X U u.uFqP-R4NK p p V UUUu/uF5wrpVP/4VP/4VP/43bK7 pppp\1VV4p\VR4;_uu_4pVP3V4RRR4\RV 24V# +'giELz;i \&d!p \RT R\)T 4 24T hRp ? ii;i +'giEL;iuup iuupppi +'giL;i)Nz def triton_(z]Need to run original Pytorch code generating kernels with TORCHINDUCTOR_UNIQUE_KERNEL_NAMES=1z.launch_paramszLaunch params file z not found. Generating...1 TORCHINDUCTOR_DUMP_LAUNCH_PARAMST.)envcapture_outputtextcwdzError running :zstdout: zstderr: zBFailed to generate launch params. Command failed with return code zSuccessfully generated z,Failed to generate launch params by running z: zMissing z1. Run `TORCHINDUCTOR_DUMP_LAUNCH_PARAMS=1 python z` first.|w)openread RuntimeErrorrrospathexistsprintenvironcopy subprocessrunsys executabledirname returncodestdoutstderr Exceptionr readlinesrEstriprOwrite)rQrRrSfilertransformed_codelaunch_params_filenamerYresulteflaunch_params_metai split_paramsabckernel_args_grids&&& r process_filerUs n  iik  #$ k  ::JK+,<= ./~> 77>>0 1 16J #$:#;;TUV **//#C69C2 3^^0#GGOON3::s F  A%~&6a890101"XY_YjYjXkl +,B+CD E 77>>0 1 1-.._`n_oow x   $ % %[[] &+==*~>NbQTUVQWPXY   & % %>V $ #sOJA,J*A>J*K,K, ;K1,K8 J' * K5KK K) 8 L ztriton_only_repro.pyc<V^8dQhR\R\R\/#)r input_path output_pathrS)rrT)r s"r r r s+ Q Q Q Q Qrc@\\V4\V4V4#)a Run experiments and output results to file Args: input_path (Path): Path to inductor generated output code output_path (Path): Path to write out the new python file auto_generate_params (bool): Whether to automatically generate launch_params if missing )rr )rrrSs&&&r get_clean_tritonrs J[)9;O PPr__main__z=Clean Inductor generated code to remove Inductor dependencies) descriptionrz&Path to inductor generated output code)typehelpz --output_pathz)Path to write out the clean triton output)rdefaultrz--no-auto-generate store_truez2Disable automatic generation of launch_params file)actionr)T)argparsercrrirkpathlibrrrr/r8rOrr__name__ArgumentParserparser add_argument parse_argsargsrrno_auto_generaterxrr rs    4$:?H34!% Q z $ $SF  4&N  +, 8    A    D ))t/D/D+DFAr