Th8 SSKrSSKrSSKrSSKrSSKJr S\S\4SjrS\S\4SjrS\S\4Sjr S\ \S \ \S\ \4S jr S \S \ \\ \\44S\4S jrS#S\S\S\S\4Sjjr\"S5S4S\S\S\4Sjjr\S:Xa\R&"SS9r\R+S\SS9 \R+S\\"S5SS9 \R+SS S!S"9 \R-5r\"\R0\R2\R4(+5rgg)$N)Path source_codereturncf[R"SSU5n[R"SSU5nU$)Nz(\n.+\s\'\'\'\n) z (\'\'\'\,.+))resub)r remove_head remove_tails W/var/www/fran/franai/venv/lib/python3.13/site-packages/torch/utils/_get_clean_triton.py"remove_triton_function_declarationr s-&&,dK@K&&$ URS5nURS5nTUup4[URS5URS55nUSUSSRU5S3nU$)Nrz, [z]())rr/splitjoin)r# func_nameparamsr*grid new_stringr1s 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)r0r1r!r transformedremove_inductor_wrapperss ` r add_launch_paramsrA7sO$G # &&84K!vv-ii   $#rTinput_filenameoutput_filenameauto_generate_paramsc [U5nUR5nSSS5 WnSU;a [S5e[U5n[ U5nUS3n[ R RU5(dU(a[SUS35 [ RR5nSUS'[R"SU/US S [ R RU5=(d S S 9nURS :waW[S US35 [SUR35 [SUR 35 [SUR35e[SU35 [ R RU5(d[SUSUS35e[U5n U R'5n SSS5 W V s/sHoR)S5PM n n U VVVs0sH5upnUR+5UR+5UR+54_M7 nnnn[-UU5n[US5nUR/U5 SSS5 [SU35 U$!,(df  GNR=f!["a n [SUS[%U 535U eSn A ff=f!,(df  N=fs sn fs snnnf!,(df  Nz=f)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_PARAMSpythonT.)envcapture_outputtextcwdrzError 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 RuntimeErrorr rospathexistsprintenvironcopy subprocessrundirname returncodestdoutstderr Exceptionr readlinesr7striprAwrite)rBrCrDfilertransformed_codelaunch_params_filenamerJresulteflaunch_params_metai split_paramsabckernel_args_grids r process_filerqTs n iik  #$ k  ::JK+,<= ./~> 77>>0 1 16J #$:#;;TUV **//#C69C2 3^^>*#GGOON3:s F  A%~&6a890101"XY_YjYjXkl +,B+CD E 77>>0 1 1-.._`n_oow x   $ %[[] &+==*~>NbQTUVQWPXY   & %>V $ #sHI(CI:'J'J8&K( I7: J$JJ$' J5 Kztriton_only_repro.py input_path output_pathc@[[U5[U5U5$)a Run experiments and output results to file Args: input_path (Optional[Path]): Path to inductor generated output codede output_path (Optional[Path]): Path to write out the new python file auto_generate_params (bool): Whether to automatically generate launch_params if missing )rqr)rrrsrDs r get_clean_tritonrus J[)9;O PPr__main__z=Clean Inductor generated code to remove Inductor dependencies) descriptionz&Path to inductor generated output code)typehelpz --output_pathz)Path to write out the clean triton output)rxdefaultryz--no-auto-generate store_truez2Disable automatic generation of launch_params file)actionry)T)argparserTrrZpathlibrrr rr(listr/dicttuplerAboolrqru__name__ArgumentParserparser add_argument parse_argsargsrrrsno_auto_generatergrr rs CC cc4$s)ctCy$$%)#uS#X*>%?$$<MQ??*-?EI??H34!% Q Q Q Q z $ $SF  4&N  +, 8    A    D ))t/D/D+DFAr