`iK ddlZddlZddlZddlZddlZddlmZddlZddlm Z ddl m Z m Z ddl mZmZmZmZddlmZejeZdedeed ed efd Zd eded efd Z ddededededed ef dZdS)N)IO)config)BuildOptionsBase CppBuilder) AOTI_FILESAOTICompiledModelload_pt2 package_pt2)FileLikeaoti_dir aoti_filesso_pathreturnc6dtdtffd }tj||d}tj||d}tj|d}t |dz5}t j|}dddn #1swxYwYtdi|dtj i} t||| } | } | t |d z5}t j|} dddn #1swxYwYtdi| dtj i} ttj|d | |g| | }| }| |d z}|vrt |d5}|}dddn #1swxYwYt |d5}|}|dd|dzz z||dddn #1swxYwY|S)NsuffixrcdD]}||r|cStd|)Nz Unable to find file with suffix )endswith RuntimeError)rfiler s s/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/torch/_inductor/package/package.pyget_aoti_file_with_suffixz-compile_so..get_aoti_file_with_suffixsK  D}}V$$   FfFFGGGz.cppz.orz_compile_flags.jsonuse_relative_path)namesources BuildOptionz_linker_flags.json)rrr output_dirz_serialized_weights.binrbza+b i@)strospathjoinsplitextopenjsonloadrr is_fbcoderget_target_file_pathbuildsplitreadtellwrite)r r rrcpp_fileconsts_o file_namef compile_flagscompile_optionsobject_builderoutput_o linker_flagslinker_options so_builder output_soserialized_weights_filename f_weightsserialized_weightsf_soso_sizes ` r compile_sorBsH#H#HHHHHHw||H&?&?&G&GHHHw||H&?&?&E&EFFH  **1-I i// 0 0%A ! %%%%%%%%%%%%%%%' +1+;+=+=O #N 2244H i.. / /$1y|| $$$$$$$$$$$$$$$& *0*:*<*<N W]]7 # #B '8$" J //11I#,.G"G"j00 -t 4 4 2 !*!1!1  2 2 2 2 2 2 2 2 2 2 2 2 2 2 2)U # # +tiikkG JJtuw67 8 8 8 JJ) * * *  + + + + + + + + + + + + + + + sICCC8EE E>HH#&H#:AJJJ archive_filec$t||S)a_ Saves the AOTInductor generated files to the PT2Archive format. Args: archive_file: The file name to save the package to. aoti_files: This can either be a singular path to a directory containing the AOTInductor files, or a dictionary mapping the model name to the path to its AOTInductor generated files. )r )r )rCr s r package_aotirERs!    rmodelFrr$ model_namerun_single_threaded num_runners device_indexc2 t||||}||jvrtd|d|j|S#t$rtdYnwxYwt |t jtfrtj d5}| d| | td|jt jj|j||||}t)|cdddS#1swxYwYt+j|}t jj|||||}t)|S) N)rIrJrKzModel z not found in packagez:Loading outdated pt2 file. Please regenerate your package.z.pt2)rrz)Writing buffer to tmp file located at %s.)r aoti_runnersrlogwarning isinstanceioIOBasertempfileNamedTemporaryFileseekr0r.debugrtorch_C_aotiAOTIModelPackageLoaderrr#fspath)r$rHrIrJrK pt2_contentsr4loaders r load_packager^fs R  3#%    \6 6 6I IIIJJ J(44 RRR PQQQQQR$B(( -  ( 7 7 7 -1 IIaLLL GGDIIKK IIA16 J J JX^:: $7lF%V,, - - - - - - - - - - - - - - - - 9T??D X^ 2 2 j-{LF V $ $$s";>$A%$A%BEE E)rFFrGr)rQr(loggingr#rStypingrrWtorch._inductorrtorch._inductor.cpp_builderrr!torch.export.pt2_archive._packagerrr r torch.typesr getLogger__name__rNr"listrBrEboolintr^r!rrrjs   """"""DDDDDDDD ! g!!88$s)8c8c8888v, % $%$% $%$%$% $%  $%  $%$%$%$%$%$%r