΅iI (SSKrSSKrSSKrSSKrSSKrSSKJr S\S\4SjrS\S\4Sjr S\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\"\R2\R4\R6(+5rgg)$N)Path source_codereturncf[R"SSU5n[R"SSU5nU$)Nz(\n.+\s\'\'\'\n) z (\'\'\'\,.+))resub)r remove_head remove_tails W/home/james-whalen/.local/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_paramsrA8sO$G # &&84K!vv-ii   $#rTinput_filenameoutput_filenameauto_generate_paramsc [U5nUR5nSSS5 WnSU;a [S5e[U5n[ U5nUS3n[ R RU5(dU(a[SUS35 [ RR5nSUS'[R"[RU/USS[ R RU5=(d S S 9nUR S :waW[S US 35 [SUR"35 [SUR$35 [SUR 35e[SU35 [ R RU5(d[SUSUS35e[U5n U R+5n SSS5 W V s/sHoR-S5PM n n U VVVs0sH5upnUR/5UR/5UR/54_M7 nnnn[1UU5n[US5nUR3U5 SSS5 [SU35 U$!,(df  GN`=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_PARAMST.)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 subprocessrunsys executabledirname returncodestdoutstderr Exceptionr readlinesr7striprAwrite)rBrCrDfilertransformed_codelaunch_params_filenamerIresulteflaunch_params_metai split_paramsabckernel_args_grids r process_filerrUs 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 $ #sHI6C&J5J5K4rs  CC cc4$s)ctCy$$%)#uS#X*>%?$$<MQ??*-?EI??H34!% Q Q Q Q z $ $SF  4&N  +, 8    A    D ))t/D/D+DFAr