o êÜÓh‘ã@s^ddlZddlmZmZmZmZmZdgZdeegefdeeedfdeefdd„Z dS) éN)ÚAnyÚCallableÚIterableÚListÚTupleÚtrace_dependenciesÚcallableÚinputs.Úreturnc sPtƒ‰‡fdd„}zt |¡|D]}||ŽqWt d¡tˆƒSt d¡w)a¾Trace the execution of a callable in order to determine which modules it uses. Args: callable: The callable to execute and trace. inputs: The input to use during tracing. The modules used by 'callable' when invoked by each set of inputs are union-ed to determine all modules used by the callable for the purpooses of packaging. Returns: A list of the names of all modules used during callable execution. csˆ|dkrdS|jj}d}||jvr|j|j}n!||jvr$|j|j}nd|jvr9t|jd|dƒ}|r7|jnd}|rBˆ |¡dSdS)NÚcallÚself)Úf_codeÚco_nameÚ f_globalsÚ __module__Úf_localsÚgetattrÚadd)ÚframeÚeventÚargÚnameÚmoduleÚmethod©Ú modules_used©ú^/var/www/html/ai/venv/lib/python3.10/site-packages/torch/package/analyze/trace_dependencies.pyÚrecord_used_moduless   ÿz/trace_dependencies..record_used_modulesN)ÚsetÚsysÚ setprofileÚlist)rr rÚinprrrrs   ÿ  þ) r ÚtypingrrrrrÚ__all__ÚstrrrrrrÚs ÿÿþ