tjzddlZddlZddlmZddlmZddlZddlZddlZddl m Z ej e Z dgZ ddejjdeedeed fd eeefdzd eeeeefff d Z dd dddejjdeed fd eeefdzdeded eeedzff dZdS)N)Iterable)Any)log_export_usagereport_exportabilitymodeltarget_submodulesargs.kwargsreturnc |pi}g}i d|D fd} |D]3\}}||vr*|||d4||i|n0#t$r#}t jd|dYd}~nd}~wwxYw|D]} | n#|D]} | wxYw S) a~ Generate inputs for targeting submdoules in the given model. Note that if two submodules refer to the same obj, this function doesn't work. Args: model: root model. inputs: inputs to the root model. target_submodules: submodules that we want to generate inputs for. Returns: A dict that maps from submodule name to its inputs. ci|]\}}|| Sr).0namemods Q/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/_export/tools.py z3_generate_inputs_for_submodules..(sKKK c#tKKKc"||f|<dSNr)module module_args module_kwargsresultssubmodule_to_namess r pre_forwardz4_generate_inputs_for_submodules..pre_forward*s/:M.J"6*+++rT) with_kwargszDFailed to generate submodule inputs because of the following error: ) stacklevelN) named_modulesappendregister_forward_pre_hook Exceptionwarningswarnremove) rrr r handlesrrrehrrs @@r_generate_inputs_for_submodulesr*s$\rFGGKKU5H5H5J5JKKKKKKKKK,,..  ID#(((11+41PP tv     WTU W W            A HHJJJJ   A HHJJJJ  Ns0ABC  B. B)$C )B..C C&TFstrict pre_dispatchrr,r-c@ td|pi}d|D}t|||| t i   fd  |d||t} D]A}|=t |dd}||Bt d t||D]} t |  S) a Report exportability issues for a module in one-shot. Args: mod: root module. args: args to the root module. kwargs: kwargs to the root module. Returns: A dict that maps from submodule name to the exception that was raised when trying to export it. `None` means the module is exportable without issue. Sample output: { '': UnsupportedOperatorException(func=), 'submod_1': UnsupportedOperatorException(func=), 'submod_2': None } zexport.report_exportability)eventc$g|] \}}|dk |S)r)rr_s r z(report_exportability..]s!NNNq42:::::rcrt|vrdSt||| tjj||| d |<t d|dS#t$rS}t| dd}t d||| |<Yd}~nd}~wwxYw| D];\}}|dkr|n|d|} |d\} } ||| | .try_exportcs <<- - - FtF||,,,  v1 ( #++!!- ,'+{#5{CCC ( ( ( GGMM$//2  > Y'({######  (#1133 L LLD&&1R&7&7dd =T=Td=T=TO)6):):** &K Jv ] K K K KsA B C$A CC$r1Nz\nrzFound %d export issues:) rr r*setvaluesr?r@r8r=rAlen)rr r r,r-all_submod_names unique_issues exceptionkeyissuerJrKrLrMs `` @@@@rrr?sS489999 \rFNNC,=,=,?,?NNN3C9I4QWXXM*,F##########JJsBf%%%EEM]]__##  y//''..q1C   c " " "KK)3}+=+=>>> E Mrr)loggingr$collections.abcrtypingrr9 torch.exporttorch.export._tracetorch._utils_internalr getLogger__name__r=__all__nnModulestrtupledictr*boolr#rrrrres$$$$$$ 222222g!! ! "%) ** 8?*}* S/* cNT ! *  #uS#X  ****`%)U  UUU U S/U cNT !U  U  U #y4  UUUUUUr