ó
    
Eñi»§  ã            	       ó¶  • S SK r S SKrS SKrS SKrS SKrS SKrS SKrS SKJr  S SK	J
r
  S SKJr  S SKJr  S SKJrJrJr  S SKJrJrJrJr  S SKJr  \ R4                  " SS	5      r\ R4                  " S
S5      r/ SQrS rS rS r S r!S r"\#\$\%\&\'" S5      \RP                  \RR                  \RT                  \RV                  4	r,S r- " S S\R\                  R^                  R`                  5      r1S r2S/S jr3 " S S5      r4\4" 5       r5  S0S jr6S0S jr7S/S jr8S r9S1S jr:S r;S r<S0S  jr=S! r>S" r?S0S# jr@S$ rAS0S% jrBS& rCS' rDS( rES) rFS* rGS+ rHS, rIS- rJS. rKg)2é    N)Ú
fake_range)Ú_find_builtin)ÚAttributeTypeIsSupportedChecker)Ú_add_script_classÚ_get_script_classÚ
_python_cu)Úget_class_propertiesÚget_default_argsÚget_jit_class_defÚget_jit_def)ÚModuleÚScriptMethodStub)Úresolution_callbackÚdef_Úoriginal_methodÚPropertyStub)r   r   )Ú_versionÚ_parametersÚ_buffersÚ_non_persistent_buffers_setÚ_backward_hooksÚ_backward_pre_hooksÚ_forward_hooksÚ_forward_hooks_with_kwargsÚ_forward_pre_hooksÚ_forward_pre_hooks_with_kwargsÚ_forward_hooks_always_calledÚ_state_dict_hooksÚ_state_dict_pre_hooksÚ_load_state_dict_pre_hooksÚ_load_state_dict_post_hooksÚ_modulesÚ_initializingÚdump_patchesc                 óü   • [        U 5      nU(       di  [        X R                  5      n[        R                  R
                  R                  U 5      n[        R                  R                  X$XQ5      n[        X5        U$ ©N)
r   r   Ú__name__ÚtorchÚjitÚfrontendÚget_default_args_for_classÚ_CÚ_jit_script_class_compiler   )ÚobjÚrcbÚqualified_nameÚscript_classÚastÚdefaultss         ÚQ/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/jit/_recursive.pyÚ_compile_and_register_classr5   5   s`   € Ü$ SÓ)€LæÜ §\¡\Ó2ˆÜ—9‘9×%Ñ%×@Ñ@ÀÓEˆÜ—x‘x×9Ñ9Ø ó
ˆô 	˜#Ô,àÐó    c                 óZ   • [         R                  " U 5      n[        XSS9n[        X#U 5      $ ©NÚRecursiveScriptModule©Ú	self_name)Ú_jit_internalÚ#createResolutionCallbackFromClosurer   r   )ÚfuncÚnamer/   r2   s       r4   Ú	make_stubr@   C   s,   € Ü
×
;Ò
;¸DÓ
A€CÜ
�dÐ,CÑ
D€CÜ˜C dÓ+Ð+r6   c                 ó\   • [        X5      n[        U[        5      (       a  U$ [        X!5      $ r&   )ÚgetattrÚ
isinstancer   r@   )Ú	nn_moduleÚmethod_namer>   s      r4   Úmake_stub_from_methodrF   I   s,   € Ü�9Ó*€DÜ�$Ô(×)Ñ)Øˆô �TÓ'Ð'r6   c                 óà   • / n[        U 5       H\  n[        XS 5      n[        R                  " U5      [        R                  R
                  L d  MB  UR                  [        X5      5        M^     U$ r&   )ÚdirrB   r<   Úget_torchscript_modifierÚFunctionModifiersÚEXPORTÚappendrF   )ÚmodÚstubsr?   Úitems       r4   Ú make_stubs_from_exported_methodsrP   X   s^   € Ø€EÜ�C–ˆÜ�s $Ó'ˆä×2Ò2°4Ó8Ü×.Ñ.×5Ñ5ô6ð �L‰LÔ.¨sÓ9Ö:ñ ð €Lr6   c                 óž   • [        U S/ 5      nS nU" [        U 5      5      n[        5       nU H  nXS;   d  M
  UR                  U5        M     U$ )NÚ__jit_ignored_attributes__c                 ó˜   • [        U 5      R                  5        VVs1 s H  u  p[        U[        5      (       d  M  UiM      snn$ s  snnf r&   )ÚvarsÚitemsrC   Úproperty)ÚmoduleÚkÚvs      r4   Úget_properties_namesÚ4jit_ignored_properties.<locals>.get_properties_namesj   s4   € Ü" 6›l×0Ñ0Ô2ÔNÒ2‘d�a´jÀÄH×6M—Ñ2ÒNÐNùÓNs
   �A¼A)rB   ÚtypeÚsetÚadd)rW   Ú!user_annotated_ignored_attributesrZ   Ú
propertiesÚuser_annoted_ignored_propertiesÚignored_attrs         r4   Újit_ignored_propertiesrc   e   sY   € Ü(/ØÐ,¨bó)Ð%òOñ &¤d¨6£lÓ3€JÜ&)£eÐ#ã9ˆØÕ%Ø+×/Ñ/°Ö=ñ :ð +Ð*r6   c                 ó`  ^ ^• [        U[        5      (       a  U$ [        U[        [        45      (       a  [        U U4S jU 5       5      $ SR	                  S [         5       5      n[        [        R                  " S[        R                  " [        U5      5       ST ST  SU S3	5      5      e)	Nc              3   ó>   >#   • U  H  n[        TUT5      v •  M     g 7fr&   )Ú_get_valid_constant)Ú.0ÚxÚattrÚ
owner_types     €€r4   Ú	<genexpr>Ú&_get_valid_constant.<locals>.<genexpr>‹   s   øé € ÐIÂqÀ!Ô(¨¨q°*×=Ð=Âqùs   ƒz, c              3   óN   #   • U  H  n[         R                  " U5      v •  M     g 7fr&   )r(   Útypename)rg   Útyps     r4   rk   rl   Œ   s   é € ÐIº°#œ%Ÿ.š.¨×-Ð-ºùs   ‚#%z

        'z' object in attribute 'Ú.zo' is not a valid constant.
        Valid constants are:
        1. a nn.ModuleList
        2. a value of type {z,}
        3. a list or tuple of (2)
        )rC   Ú_constant_typesÚtupleÚlistÚjoinÚ	TypeErrorÚtextwrapÚdedentr(   rn   r\   )ri   rY   rj   Ú	constantss   ` ` r4   rf   rf   ‡   s    ù€ Ü�!”_×%Ñ%ØˆÜ	�Aœœt�}×	%Ñ	%ÜÕIÁqÓIÓIÐIØ—	‘	ÑI½ÓIÓI€IÜ
Ü�Šð
Ü
�.Š.œ˜a›Ó
!Ð	"Ð"9¸*¸ÀQÀtÀfð Mð (˜[ð )	ð	ó	
ó
ð 
r6   c                   ó   • \ rS rSrSrg)ÚSourceContextéš   © N)r'   Ú
__module__Ú__qualname__Ú__firstlineno__Ú__static_attributes__r|   r6   r4   rz   rz   š   s   † Úr6   rz   c                 ó¢   ^• [         R                  " U 5      nU(       a  U$ U4S jm[        U [        5      (       a  U O
[        U 5      nT" U5      $ )Nc                 ó’   >• [         R                  " U 5      nU(       a  U$ U R                   H  nT" U5      nU(       d  M  Us  $    0 $ r&   )ÚinspectÚget_annotationsÚ	__bases__)ÚclsÚcls_annotationsÚbaseÚget_cls_annotationss      €r4   r‰   Ú,get_annotations.<locals>.get_cls_annotations§   sF   ø€ Ü!×1Ò1°#Ó6ˆÞØ"Ð"Ø—M”MˆDÙ1°$Ó7ˆOßˆØ&Ò&ñ "ð ˆ	r6   )rƒ   r„   rC   r\   )r.   Úannotationsr†   r‰   s      @r4   r„   r„   ž   sG   ø€ ô
 ×)Ò)¨#Ó.€KÞØÐõô ˜C¤×&Ñ&‰#¬D°«I€CÙ˜sÓ#Ð#r6   c                 ó°  ^• [         R                  R                  [        U 5      5      n[	        U [         R
                  R                  5      (       a  UR                  5         [	        U [         R
                  R                  [         R
                  R                  45      (       a  UR                  5         [	        U [         R
                  R                  5      (       a  UR                  5         [	        U [         R
                  R                  5      (       a  UR                  5         [        U 5      m[	        U [         R                   R"                  R$                  5      (       a  0 m['        U S/ 5      nUR)                  U5        [+        U 5      nU4S jn[-        5       nU R.                  R1                  5        H„  u  pxXs;   a  M  Ub9  [	        U[         R2                  5      (       d  [5        SU S[        U5       35      eU" Xx5      u  pšUR7                  XyR                  5       SS5        UR9                  U5        M†     U R:                  R1                  5        H„  u  pxXs;   a  M  Ub9  [	        U[         R2                  5      (       d  [5        SU S[        U5       35      eU" Xx5      u  pšUR7                  XyR                  5       SS5        UR9                  U5        M†     U R<                  R1                  5        Hè  u  pxXs;   a  M  U" Xx5      u  pšUc#  UR7                  XyR                  5       SS5        M<  U	R?                  5       (       aj  U	R                  5       RA                  5       (       d  [5        S	U S
35      e[         R                  RB                  RE                  U	R                  5       5      nO[G        X�5      nURI                  X{5        UR9                  U5        Mê     [-        ['        U SS5      5      nTR1                  5        H<  u  p}[         RJ                  RM                  U5      (       d  M+  UR9                  U5        M>     U HÝ  nXv;   a_  XpR<                  ;   a  SnO/XpR:                  ;   a  SnOXpR.                  ;   a  SnO[5        S5      e[N        RP                  " S
U SU S3SS9  Mg  [S        X5      (       d  [N        RP                  " S
U S3SS9  M’  ['        X5      nURU                  U[W        X[        U 5      RX                  5      5        UR9                  U5        Mß     ['        U S0 5      nUR[                  []        [_        X5      5      5        UR1                  5        H  u  nnURa                  UU5        M     U Rb                  R1                  5        GH  u  pU[d        ;   d  URg                  S5      (       a  M(  Xs;   a  M/  Xv;   a  M6  [	        U[         Rh                  Rj                  5      nU(       a  URl                  n[n        Rp                  " U5      (       a`   [         Rr                  Ru                  U5      nURw                  U[         R                  Ry                  U5      R                  5       U5        Mè  [        U5      nU(       a  UR�                  UU5        GM  [	        U[         Rr                  R‚                  5      (       aA  URw                  U[         R                  Ry                  U5      R                  5       U5        GMy  U" X5      u  n	nU	R?                  5       (       a$  UR7                  XyR                  5       SS5        GM½  U(       a  SOSnU	R…                  5        SU 3nS[         R†                  " [        U5      5       SU S3nUR}                  X~5        GM     U Rˆ                  R‹                  5        H  nUR�                  U5        M     U RŽ                  R‹                  5        H  nUR‘                  U5        M     U$ ! [z         a"  nSU S3nUR}                  X~5         SnAGNÀSnAff = f)z£
Build a ConcreteModuleTypeBuilder from an nn.Module.

This ConcreteModuleType doesn't have a JIT type associated with it yet, it
must be filled in by the caller.
rR   c           	      óÆ  >• Sn U T;   a…  TU    [         R                  R                  R                  S   :w  aW  [         R                  R
                  R                  TU    [        5       5      n[         R                  R                  U5      nXB4$ [        U[         R                  R                  5      (       a^  [         R                  R
                  R                  UR                  [        5       5      n[         R                  R                  U5      nXB4$ [         R                  R                  U5      nSn XB4$ ! [         a  n[        SU  SU SU 35      UeS nAff = f)NFÚforwardTzError inferring type for z: )r(   Únnr   Ú__annotations__r)   r‹   Úann_to_typer   r,   ÚInferredTyperC   Ú	Attributer\   Ú_jit_try_infer_typeÚRuntimeError)r?   rO   Úinferredr‘   Ú	attr_typeÚreÚclass_annotationss         €r4   Ú
infer_typeÚ/infer_concrete_type_builder.<locals>.infer_typeÒ   s;  ø€ ð ˆð	YàÐ)Ó)Ø% dÑ+Ü—8‘8—?‘?×2Ñ2°9Ñ=ó>ô $Ÿi™i×3Ñ3×?Ñ?Ø% dÑ+¬Z«\ó�ô "ŸH™H×1Ñ1°+Ó>�	ð Ð"Ð"ô ˜D¤%§)¡)×"5Ñ"5×6Ñ6Ü#Ÿi™i×3Ñ3×?Ñ?ÀÇ	Á	Ì:Ë<ÓX�Ü!ŸH™H×1Ñ1°+Ó>�	ð Ð"Ð"ô "ŸH™H×8Ñ8¸Ó>�	Ø‘ð Ð"Ð"øô ó 	YÜÐ!:¸4¸&ÀÀ4À&ÈÈ2È$ÐOÓPÐVXÐXûð	Yús%   …BD< ÂBD< Ä!D< Ä<
E ÅEÅE NzExpected parameter 'z' to be None or Tensor, got TFzExpected buffer 'z1Expected inferred type to be interface type for 'Ú'Ú__constants__r|   Ú	submoduleÚbufferÚ	parameterz3added_names must be submodule, parameter, or bufferzA' was found in ScriptModule constants,  but it is a non-constant z. Consider removing it.é   )Ú
stacklevelzb' was found in ScriptModule constants, but was not actually set in __init__. Consider removing it.Ú__overloads__Ú__z˜(This function exists as an attribute on the Python module, but we failed to compile it to a TorchScript function. 
The error stack is reproduced here:
Ú)zFIts type was inferred; try adding a type annotation for the attribute.Ú z. zT(This attribute exists on the Python module, but we failed to convert Python type: 'z' to a TorchScript type. )Ir(   r,   ÚConcreteModuleTypeBuilderr\   rC   r�   Ú
ModuleDictÚset_module_dictÚ
ModuleListÚ
SequentialÚset_module_listÚParameterListÚset_parameter_listÚParameterDictÚset_parameter_dictr„   ÚaoÚquantizationÚQuantWrapperrB   Úadd_ignored_attributesrc   r]   r   rU   ÚTensorÚAssertionErrorÚadd_attributer^   r   r"   ÚsuccessÚis_interface_typeÚConcreteModuleTypeÚfrom_jit_typeÚget_module_concrete_typeÚ
add_moduler<   Úis_finalÚwarningsÚwarnÚhasattrÚadd_constantrf   r'   ÚupdateÚget_overload_name_mappingÚget_overload_annotationsÚadd_overloadÚ__dict__Úignored_attributesÚ
startswithÚ_opsÚOpOverloadPacketÚoprƒ   Ú
isfunctionr)   ÚscriptÚadd_function_attributer”   Ú	ExceptionÚadd_failed_attributer   Úadd_builtin_functionÚScriptFunctionÚreasonrn   r   ÚvaluesÚadd_forward_hookr   Úadd_forward_pre_hook)rD   Úshare_typesÚconcrete_type_builderr_   Úignored_propertiesrš   Úadded_namesr?   rO   r—   Ú_Úsub_concrete_typeÚconstants_setÚannÚhintÚvalueÚ	overloadsÚoverloaded_namesÚisoverloadpacketÚscripted_fnÚeÚbuiltin_symbol_namer–   Úinferred_msgÚadditional_infoÚhookÚpre_hookr™   s                              @r4   Úinfer_concrete_type_builderrì   µ   sE  ø€ ô "ŸH™H×>Ñ>¼tÀI»ÓOÐÜ�)œeŸh™h×1Ñ1×3Ñ3Ø×-Ñ-Ô/Ü�)œeŸh™h×1Ñ1´5·8±8×3FÑ3FÐG×HÑHØ×-Ñ-Ô/Ü�)œeŸh™h×4Ñ4×6Ñ6Ø×0Ñ0Ô2Ü�)œeŸh™h×4Ñ4×6Ñ6Ø×0Ñ0Ô2ä'¨	Ó2ÐÜ�)œeŸh™h×3Ñ3×@Ñ@×BÑBØÐô )0ØÐ/°ó)Ð%ð ×0Ñ0Ð1RÔSÜ/°	Ó:Ðõ#ô8 “%€Kà×+Ñ+×1Ñ1Ö3‰
ˆØÓ4Ùà‘¤
¨4´·±× >Ñ >Ü Ø& t fÐ,HÌÈdËÈÐUóð ñ " $Ó-‰ˆ	ð 	×+Ñ+¨D·.±.Ó2BÀDÈ%ÔPØ�‰˜Öñ 4ð"  ×(Ñ(×.Ñ.Ö0‰
ˆØÓ4Ùà‘¤
¨4´·±× >Ñ >Ü Ø# D 6Ð)EÄdÈ4ÃjÀ\ÐRóð ñ " $Ó-‰ˆ	Ø×+Ñ+¨D·.±.Ó2BÀEÈ4ÔPØ�‰˜Öñ 1ð  ×(Ñ(×.Ñ.Ö0‰
ˆØÓ4Ùá! $Ó-‰ˆ	Ø‰<ð "×/Ñ/°·n±nÓ6FÈÈuÔUÙØ×Ñ×ÑØ—>‘>Ó#×5Ñ5×7Ñ7Ü$ØGÈÀvÈQÐOóð ô !&§¡× ;Ñ ;× IÑ IØ—‘Ó ó!Ñô
 !9¸Ó KÐØ×(Ñ(¨ÔAà�‰˜Öñ1 1ô6 œ 	¨?¸BÓ?Ó@€Mð '×,Ñ,Ö.‰	ˆÜ×Ñ×'Ñ'¨×,Ó,Ø×Ñ˜dÖ#ñ /ó ˆØÓð ×)Ñ)Ó)Ø"‘Ø×+Ñ+Ó+Ø‘Ø×.Ñ.Ó.Ø"‘ä$ØIóð ô �MŠMØ�D�6ð -Ø-1¨FÐ2IðKàòñ
 Ü�y×'Ñ'ô �MŠMØ�D�6ð (ð (ð ò	ñ Ü˜	Ó(ˆØ×*Ñ*ØÔ% d´4¸	³?×3KÑ3KÓLô	
ð 	�‰˜ÖñG ôL ˜	 ?°BÓ7€Ià×ÑÜ!Ü$ YÓCó	
ôð
 #,§/¡/Ö"3ÑˆÐØ×*Ñ*¨4Ð1AÖBñ #4ð !×)Ñ)×/Ñ/×1‰ˆØÔ%Ó%¨¯©¸×)>Ñ)>ñ àÓ4ÙàÓáä% e¬U¯Z©Z×-HÑ-HÓIÐÞØ—H‘HˆEä×Ò˜e×$Ñ$ðGÜ#Ÿi™i×.Ñ.¨uÓ5�Ø%×<Ñ<Øœ%Ÿ(™(×6Ñ6°{ÓC×HÑHÓJÈEôñ ô ,¨EÓ2ÐÞØ!×6Ñ6°tÐ=PÔQÚô �eœUŸY™Y×5Ñ5×6Ñ6Ø!×8Ñ8Ø”e—h‘h×2Ñ2°5Ó9×>Ñ>Ó@À%ôò ñ )¨Ó5Ñˆ	�8Ø×Ñ×ÑØ!×/Ñ/°·n±nÓ6FÈÈu×Uö ñ Yàð ð
 "+×!1Ñ!1Ó!3Ð 4°B°|°nÐEˆOð:Ü:?¿.º.ÌÈeËÓ:UÐ9Vð W*Ø*9Ð):¸!ð=ð ð
 "×6Ñ6°t×BñI 2ðN ×(Ñ(×/Ñ/Ö1ˆØ×.Ñ.¨tÖ4ñ 2à×0Ñ0×7Ñ7Ö9ˆØ×2Ñ2°8Ö<ñ :ð !Ð øôk ó 	Gð
>à>?¸SÀðCð ð
 &×:Ñ:¸4×FÒFûð	Gús   Ø3A`)à)
aà3aáac                   ó¦   • \ rS rSr% \\\   \\R                  R                     4   \S'   \\R                  R                     \S'   SS jrS rSrg)	ÚConcreteTypeStorei®  Ú
type_storeÚmethods_compiledNc                 ó0   • 0 U l         [        5       U l        g r&   )rï   r]   rð   )Úselfs    r4   Ú__init__ÚConcreteTypeStore.__init__²  s   € àˆŒä #£ˆÕr6   c                 ó.  • [        U5      n[        U5      nX0R                  ;  a  / U R                  U'   U R                  U   nU H  nUR                  U5      (       d  M  Us  $    UR	                  5       nU R                  U   R                  U5        U$ )zaInfer a ConcreteType from this `nn.Module` instance. Underlying JIT types are reused if possible.)rì   r\   rï   ÚequalsÚbuildrL   )rò   rD   rÙ   Únn_module_typeÚknown_typesÚ
known_typeÚconcrete_types          r4   Úget_or_create_concrete_typeÚ-ConcreteTypeStore.get_or_create_concrete_type¸  s�   € ä ;¸IÓ FÐä˜i›ˆØ§¡Ó0Ø.0ˆD�O‰O˜NÑ+ð —o‘o nÑ5ˆÛ%ˆJØ× Ñ Ð!6×7Ó7Ø!Ò!ñ &ð
 .×3Ñ3Ó5ˆØ�‰˜Ñ'×.Ñ.¨}Ô=ØÐr6   )rð   rï   ©ÚreturnN)r'   r}   r~   r   Údictr\   r   rs   r(   r,   rº   r�   r]   ró   rü   r€   r|   r6   r4   rî   rî   ®  sD   ‡ Ø�T˜&‘\ 4¨¯©×(CÑ(CÑ#DÐDÑEÓEØ˜%Ÿ(™(×5Ñ5Ñ6Ó6ô&õr6   rî   c                 ó~  • U Vs/ s H  o3R                   PM     nnU Vs/ s H  o3R                  PM     nnU Vs/ s H  n[        UR                  5      PM     nnU Vs/ s H  owR                   PM     nnU Vs/ s H  owR                  PM     n	nU R	                  X‰XEU5        g s  snf s  snf s  snf s  snf s  snf r&   )r   r   r
   r   Ú_create_methods_and_properties)
rû   Úmethod_stubsÚproperty_stubsÚmÚmethod_defsÚmethod_rcbsÚmethod_defaultsÚpÚproperty_defsÚproperty_rcbss
             r4   Ú(create_methods_and_properties_from_stubsr  Ï  s¦   € ñ $0Ó0¢<˜a—6”6¡<€KÐ0Ù2>Ó?²,¨Q×(Ô(±,€KÐ?ÙDPÓQÂL¸qÔ'¨×(9Ñ(9Ö:ÁL€OÐQá%3Ó4¢^ —V”V¡^€MÐ4Ù4BÓC²N¨q×*Ô*±N€MÐCà×0Ñ0Ø kÀõùò 1ùÚ?ùÚQùâ4ùÚCs   …B& B+»B0Á B5Á;B:c                 ó(  • U Vs/ s H  o3R                   PM     nnU Vs/ s H  o3R                  PM     nnU Vs/ s H  o3R                   PM     nnU Vs/ s H  o3R                  PM     nnU R                  XEXg5        g s  snf s  snf s  snf s  snf r&   )r   r   Ú_create_hooks)rû   Ú
hook_stubsÚpre_hook_stubsÚhÚ	hook_defsÚ	hook_rcbsÚpre_hook_defsÚpre_hook_rcbss           r4   Úcreate_hooks_from_stubsr  Þ  s|   € Ù!+Ó,¢˜A—”¡€IÐ,Ù0:Ó;²
¨1×&Ô&±
€IÐ;á%3Ó4¢^ —V”V¡^€MÐ4Ù4BÓC²N¨q×*Ô*±N€MÐCà×Ñ 	°mÕSùò -ùÚ;ùâ4ùÚCs   …B  B»B
ÁBc                 ó|  • [        U [        5      (       d  [        S[        U 5       35      e[        U [        R
                  R                  5      (       a  [        U S5      (       a  U R                  $ U(       a  [        R                  U 5      nU$ [        X5      nUR                  5         UR                  5       nU$ )a³  
Get a concrete type for nn_modules.

If share_types is True, the concrete type is fetched from concrete_type_store.
If it is False, a new concrete type is created without first searching concrete_type_store.

Args:
    nn_module:  The original Python nn.Module that we are creating a ScriptModule for.
    share_types = Whether to share underlying JIT types between modules (if possible).

Returns:
    A concrete type for nn_module.
zExpected Module, got Ú_concrete_type)rC   r   r¶   r\   r(   r)   ÚScriptModulerÁ   r  Úconcrete_type_storerü   rì   Úset_poisonedr÷   )rD   rØ   rû   rÙ   s       r4   r¼   r¼   è  s¨   € ô �i¤×(Ñ(ÜÐ4´T¸)³_Ð4EÐFÓGÐGÜ�)œUŸY™Y×3Ñ3×4Ñ4¼ØÐ#÷:ñ :ð ×'Ñ'Ð'æä+×GÑGÈ	ÓRˆð Ðô	 !<¸IÓ SÐØ×*Ñ*Ô,Ø-×3Ñ3Ó5ˆàÐr6   c                 ó�  • [         R                  " [        U 5      5      n[         R                  " [        U 5      5      n[	        [        U 5      X!5        [
        R                  " U5      n[        R                  R                  U5      nU R                  R                  5        H  u  pVUR                  XV5        M     [        U5      $ )zo
Create and return a RecursiveScriptClass instance from a Python object.

Arguments:
    obj: A Python object.
)r<   Ú_qualified_namer\   Ú'createResolutionCallbackForClassMethodsr5   r   Ú	get_classr(   r,   Ú_create_object_with_typerÇ   rU   ÚsetattrÚwrap_cpp_class)r.   Úqualified_class_namer/   Úclass_tyÚ
cpp_objectr?   rá   s          r4   Úcreate_script_classr&  
  s”   € ô )×8Ò8¼¸c»ÓCÐÜ
×
?Ò
?ÄÀSÃ	Ó
J€Cä¤ S£	¨3ÔEÜ×#Ò#Ð$8Ó9€Hä—‘×2Ñ2°8Ó<€Jà—|‘|×)Ñ)Ö+‰ˆØ×Ñ˜4Ö'ñ ,ô ˜*Ó%Ð%r6   c                 óî   • [        U [        R                  R                  5      (       a  [	        S5      e[        U 5        [        X5      nU(       d  [        5       R                  U 5        [        XU5      $ )a¨  
Create a new ScriptModule from an nn.Module.

Args:
    nn_module:  The original Python nn.Module that we are creating a ScriptModule for.
    stubs_fn:  Lambda that takes an nn.Module and generates a list of ScriptMethodStubs to compile.
    share_types:  Whether to share underlying JIT types between modules (if possible).
        NOTE: Only set to False this when we cannot guarantee type sharing will work
            correctly. This only happens today for traced modules, where the same
            module can produce different traced methods depending on the inputs.
    is_tracing: Whether this function is called during tracing or scripting. If tracing,
            we don't need to do AttributeTypeIsSupportedChecker because all the unsupported
            attributes will be baked as constant in the tracing graph. In addition,
            this check significantly slows down the traced modules when the module size is big.
z8Cannot script a RecursiveScriptModule (already compiled))
rC   r(   r)   r9   r¶   Úcheck_module_initializedr¼   r   ÚcheckÚcreate_script_module_impl)rD   Ústubs_fnrØ   Ú
is_tracingrû   s        r4   Úcreate_script_moduler-     sZ   € ô  �)œUŸY™Y×<Ñ<×=Ñ=ÜÐWÓXÐXÜ˜YÔ'Ü,¨YÓD€MÞÜ'Ó)×/Ñ/°	Ô:Ü$ Y¸xÓHÐHr6   c                 óº  ^ ^^^^• [         R                  R                  TR                  5      mT" T 5      n[	        T 5      n[        T 5      u  pV[        T 5      mS	UUUU U4S jjn[         R                  R                  R                  TU5      nT[        R                  ;  aV  [        TX45        [        TXV5        [         R                  R                  T5        [        R                  R                  T5        [!        UR"                  R%                  5       5       H  u  pšX¨R&                  U	'   M     [!        UR"                  R)                  5       5       H  u  pšX¨R*                  U	'   M     [-        T [         R.                  R0                  [         R.                  R2                  [         R.                  R4                  45      (       a2  STR7                  5       ;  a  UR9                  S[;        T 5       S35        [-        T [         R.                  R4                  5      (       a{  STR7                  5       ;  ag  [;        T R=                  5       5      (       a8  [?        [A        T R=                  5       5      5      nUR9                  SU S35        OUR9                  S5        U H–  nURB                  c  M  URB                  RD                  nXÜRF                  RI                  5       RH                  :w  a  MQ  TRK                  U5      n[L        RN                  " URB                  5      " U5      nXøRP                  U'   M˜     U H²  nURF                  RI                  5       RH                  nTRK                  URF                  RS                  5       RH                  5      nURF                  RU                  5       nU(       a  TRK                  URH                  5      OSn[W        UUU5      URP                  U'   M´     [Y        T 5       HX  nUT;   a  M  [[        T US5      n[\        R^                  " U5      [\        R`                  Rb                  L d  MK  [e        UT U5        MZ     U$ )
a:  
Convert an nn.Module to a RecursiveScriptModule.

Args:
    nn_module:  The original Python nn.Module that we are creating a ScriptModule for.
    concrete_type:  The fully initialized ConcreteType of the module.
    stubs_fn:  Lambda that takes an nn.Module and generates a list of ScriptMethodStubs to compile.
Nc                 ó(  >• T	R                  5        HW  n[        TU5      n[        U[        R                  R
                  5      (       a  UR                  OUnT
R                  X5        MY     T	R                  5        HÕ  u  p[        TU5      n[        U[        5      (       d  [        S[        U5       35      eUR                  n[        U[        R                  R                  5      (       a  [        XB5      nO8[        U[        R                  R                   5      (       a  UnO[#        X#T5      nT
R                  X5        XPR$                  U'   M×     ['        T5       H§  nUT;   a  M  [        TUS 5      n[(        R*                  " U5      (       aP  [,        R.                  " U5      (       a5  [        TU5      R0                  nUR3                  U 5      n[        XU5        Mƒ  T	R5                  U5      (       d  M›  [        XU5        M©     T	U l        g )NzExpected Module but got )Úget_attributesrB   rC   r(   r)   r“   rá   r!  Úget_modulesr   r¶   r\   Újit_typer,   ÚInterfaceTypeÚinterface_scriptr  r*  r"   rH   rƒ   Úismethodr<   Úis_ignored_fnÚ__func__Ú__get__Úis_ignored_attributer  )Úscript_moduler?   Ú
orig_valuerÝ   Úmodule_typeÚscriptedrO   Úunbound_functionÚbound_methodrû   Ú
cpp_modulerÚ   rD   r+  s            €€€€€r4   Úinit_fnÚ*create_script_module_impl.<locals>.init_fnH  s¿  ø€ ð "×0Ñ0Ö2ˆDÜ  ¨DÓ1ˆJô ˜j¬%¯)©)×*=Ñ*=×>Ñ>ð × Ò àð ð
 ×Ñ˜tÖ0ñ 3ð (5×'@Ñ'@Ö'BÑ#ˆDÜ  ¨DÓ1ˆJÜ˜j¬&×1Ñ1Ü$Ð'?ÄÀZÓ@PÐ?QÐ%RÓSÐSØ+×4Ñ4ˆKÜ˜+¤u§x¡x×'=Ñ'=×>Ñ>ä+¨KÓD‘Ü˜J¬¯	©	×(>Ñ(>×?Ñ?Ø%‘ô 5Ø°8ó�ð ×Ñ˜tÔ.Ø+3×"Ñ" 4Ó(ñ# (Cô* ˜	–NˆDØÐ)Ó)ÙÜ˜9 d¨DÓ1ˆDÜ×Ò ×%Ñ%¬-×*EÒ*EÀd×*KÑ*KÜ#*¨9°dÓ#;×#DÑ#DÐ Ø/×7Ñ7¸ÓF�Ü˜¨\Ö:Ø×3Ñ3°D×9Ó9Ü˜¨TÖ2ñ #ð (5ˆÕ$r6   Ú__len__zdef __len__(self):
   return Ú
Ú__contains__z3def __contains__(self, key: str):
   return key in z2def __contains__(self, key: str):
   return False
rþ   )3r(   r,   Ú_create_module_with_typer2  Úget_property_stubsÚget_hook_stubsrc   r)   r9   Ú
_constructr  rð   r  r  Ú_run_emit_module_hookr^   Ú	enumerateÚ_cÚ_get_forward_pre_hooksr   Ú_get_forward_hooksr   rC   r�   rª   r«   r¨   Ú_method_namesÚdefineÚlenÚkeysÚreprrs   r   r'   r   r?   Ú_get_methodÚ	functoolsÚwrapsrÇ   Úgetter_nameÚsetter_namerV   rH   rB   r<   rI   rJ   ÚCOPY_TO_SCRIPT_WRAPPERÚ!add_python_attr_to_scripted_model)rD   rû   r+  r  r  r  r  rA  r:  ÚidxÚfnrR  Úmethod_stubr?   Úscript_methodÚwrapped_script_methodÚproperty_stubÚproperty_nameÚfgetrX  ÚfsetrO   r@  rÚ   s   ```                   @@r4   r*  r*  9  sŠ  ü€ ô —‘×2Ñ2°=×3IÑ3IÓJ€JÙ˜IÓ&€LÜ'¨	Ó2€NÜ!/°	Ó!:Ñ€JÜ/°	Ó:Ð÷/5ó /5ôd —I‘I×3Ñ3×>Ñ>¸zÈ7ÓS€Mð Ô/×@Ñ@Ó@Ü0Ø˜<ô	
ô
 	  ¨zÔJÜ�‰×&Ñ& zÔ2Ü×,Ñ,×0Ñ0°Ô?ô ˜]×-Ñ-×DÑDÓFÖG‰ˆØ02×(Ñ(¨Ó-ñ Hä˜]×-Ñ-×@Ñ@ÓBÖC‰ˆØ,.×$Ñ$ SÓ)ñ Dô
 	ØœŸ™×+Ñ+¬U¯X©X×-@Ñ-@Ä%Ç(Á(×BUÑBUÐV÷	
ñ 	
ð ˜Z×5Ñ5Ó7Ó7à×ÑÐ=¼cÀ)»nÐ=MÈRÐPÔQä�9œeŸh™h×1Ñ1×2Ñ2Ø *×":Ñ":Ó"<Ó<äˆy�~‰~Ó× Ñ Üœ˜YŸ^™^Ó-Ó.Ó/ˆDØ× Ñ ØFÀtÀfÈBÐOõð × Ñ Ð!WÔXó $ˆØ×&Ñ&Ñ.ñ à×*Ñ*×3Ñ3ˆØ×#Ñ#×(Ñ(Ó*×/Ñ/Ó/ñ Ø"×.Ñ.¨tÓ4ˆô
 !*§¢°×0KÑ0KÔ LØó!
Ðð (=×Ñ˜tÓ$ñ/ $ó4 (ˆØ%×*Ñ*×/Ñ/Ó1×6Ñ6ˆØ×%Ñ% m×&8Ñ&8×&DÑ&DÓ&F×&KÑ&KÓLˆà#×(Ñ(×4Ñ4Ó6ˆÞ;Fˆz×%Ñ% k×&6Ñ&6Ô7ÈDˆÜ08¸ÈÈdÓ0Sˆ×Ñ˜}Ó-ñ (ô �I–ˆØÐ%Ó%ÙÜ�y $¨Ó-ˆä×2Ò2°4Ó8Ü×.Ñ.×EÑEôFô .¨m¸YÈÖMñ ð Ðr6   c                 ó|   • [        XS 5      nUc  g[        [        R                  R                  US 5      nUc  gX#:g  $ )NF)rB   r(   r)   r9   )Úscript_modelri   Úscript_attrÚdefault_attrs       r4   Úscript_model_defines_attrrh  ×  sA   € Ü˜,¨dÓ3€KØÑØÜœ5Ÿ9™9×:Ñ:¸DÀ$ÓG€LØÑØØÑ&Ð&r6   c                 ór   • [        X5      (       a'  [        X5      (       a  [        X[        X5      5        g g g r&   )rÁ   rh  r!  rB   )re  Úorigri   s      r4   rZ  rZ  á  s0   € Üˆt×ÑÔ8¸×LÑLÜ�¤G¨DÓ$7Õ8ð  MÐr6   c                 ó&  • 0 n[        [        U 5      5       Hñ  nX1;   a  M
  [        XS 5      n[        U5      (       d  M(  [	        US5      (       d  M;  UR
                  c  MJ  [        R                  " X@R                  5      nUc  Mo  UR                  U;   a*  [        [        R                  " SUR                  5      5      e[        [        U5      5       Vs/ s H  ocS-   [        U5      -   PM     nn[        [!        Xu5      5      X$'   Mó     U$ s  snf )Nr}   Úmethodr¤   )rH   r\   rB   ÚcallablerÁ   r}   r<   Ú_get_overloaded_methodsÚ	__class__r7  r•   Ú,get_overload_no_implementation_error_messageÚrangerQ  Ústrrs   Úzip)rM   rc   râ   r?   rO   Úmethod_overloadsÚiÚnamess           r4   rÅ   rÅ   æ  sõ   € à€Iä”D˜“I–ˆØÓ)ÙÜ�s $Ó'ˆÜ˜�~‰~Ùô �4˜×&Ó&¨4¯?©?Ó+FÜ,×DÒDØ—m‘mó Ðð  Ñ'Ùð �}‰}Ð 0Ó0Ü"Ü!×NÒNØ  $§-¡-óóð ô 49¼Ð=MÓ9NÔ3OÓPÒ3O¨a˜D‘[¤3 q£6Ô)Ñ3OˆEÐPÜ"¤3 uÓ#?Ó@ˆI‹Oñ1 ð4 Ðùò Qs   ÃDc                 ó¢   • 0 nU R                  5        H8  u  p#UR                  nXA;  a  / X'   U H  u  pVX   R                  U5        M     M:     U$ r&   )rU   r'   rL   )Úoverload_infoÚoverload_name_mappingsÚorig_fnrâ   Úoriginal_nameÚoverload_namerÜ   s          r4   rÄ   rÄ     s`   € ð 46ÐØ+×1Ñ1Ö3ÑˆØ×(Ñ(ˆØÓ6Ø46Ð"Ñ1ã )ÑˆMØ"Ñ1×8Ñ8¸ÖGó !*ñ 4ð "Ð!r6   c                 óâ   • [         R                  R                  R                  U S [	        5       [
        R                  " U 5      5      nUc$  [        R                  " U 5      n[        SU 35      eg )Nz>Must explicitly add type annotations to overloaded functions: )
r(   r)   r‹   Úget_signaturer   rƒ   r5  r<   r  r•   )r>   Ú	signatureÚ	qual_names      r4   Ú_check_no_signaturer�    sf   € Ü—	‘	×%Ñ%×3Ñ3Øˆd”J“L¤'×"2Ò"2°4Ó"8ó€Ið ÑÜ!×1Ò1°$Ó7ˆ	ÜØLÈYÈKÐXó
ð 	
ð r6   c           	      óv  • / nU R                  5        H¢  u  p#[        X"R                  SS9nU Hƒ  u  pV[        U5        [        XfR                  SS9n[        R
                  R                  UR                  5       XE5      n[        R                  " U5      n	UR                  [        X˜U5      5        M…     M¤     U$ r8   )rU   r   r'   r�  r(   r,   Ú_replace_overloaded_method_declÚdeclr<   r=   rL   r   )
rx  Úoverload_stubsrz  râ   Úorig_astr|  Úoverload_fnÚover_astÚnew_astÚ_rcbs
             r4   Úmake_stubs_for_overloadsr‹     s±   € Ø€NØ+×1Ñ1Ö3ÑˆÜØ×%Ñ%Ð1Hñ
ˆó +4Ñ&ˆMÜ Ô,Ü"Ø×1Ñ1Ð=TñˆHô —h‘h×>Ñ>Ø—‘“ óˆGô !×DÒDÀWÓMˆDØ×!Ñ!Ô"2°4À+Ó"NÖOó +4ñ	 4ð Ðr6   c           	      ó8  • [        U [        R                  R                  5      (       d  [	        S[        U 5       35      e[        U S5      (       d,  [        S[        R                  " [        U 5      5       S35      e[        U S5      (       Gd  U R                  R                  5        Hg  u  pUc  M
  [        R                  R                  R                  U5      (       d  M:  [        S[        R                  " [        U 5      5       SU S35      e   U R                  R                  5        Hg  u  pUc  M
  [        R                  R                  R                  U5      (       d  M:  [        S[        R                  " [        U 5      5       SU S35      e   g g )	NzExpected torch.nn.Module, got r   rœ   z=' has not been initialized, did you forget to call 'super()'?Úremote_parametersz' has uninitialized parameters z'. Did you forget to run a forward pass?z' has uninitialized buffers )rC   r(   r�   r   r¶   r\   rÁ   r•   rn   r   rU   r    Úis_lazyr   )rM   r?   ÚparamÚbufs       r4   r(  r(  3  sK  € Ü�cœ5Ÿ8™8Ÿ?™?×+Ñ+ÜÐ=¼dÀ3»i¸[ÐIÓJÐJÜ�3˜×&Ñ&ÜØ”—’œt C›yÓ)Ð*Ð*gÐhó
ð 	
ô
 �3Ð+×,Ò,ØŸ?™?×0Ñ0Ö2‰KˆDØÓ ¤U§X¡X×%7Ñ%7×%?Ñ%?À×%FÓ%FÜ"ØœŸš¤t¨C£yÓ1Ð2Ð2QÐRVÐQWÐW~Ðóð ñ 3ð
 Ÿ™×+Ñ+Ö-‰IˆDØ‹¤5§8¡8×#5Ñ#5×#=Ñ#=¸c×#BÓ#BÜ"ØœŸš¤t¨C£yÓ1Ð2Ð2NÈtÈfÐT{Ð|óð ò .ð -r6   c                 ó†  ^• [        U 5        [        U 5      n/ n[        U S5      (       ai  [        R                  " U R
                  5      (       dD  [        U R
                  SS5      n[        [        R                  R                  SS5      nX4:w  a  S/n/ n[        U 5       HZ  nXa;   a  M
  [        XS5      n[        R                  " U5      [        R                  R                  L d  MI  UR                  U5        M\     X%-   n[        [        U S0 5      5      m[!        X5      nTR#                  [%        U5      5        ['        U5      n	TU l        U4S jn
[+        X¢5      n[-        5       n/ nU H,  nXl;   a  M
  UR                  U5        UR/                  U5        M.     U Vs/ s H  n[1        X5      PM     nnXŸ-   $ s  snf )zŽImplement the default rules for which methods should act as starting points for compilation.

(TODO add a link when the rules are published).
rŽ   r7  Nr£   c                 ó   >• U T;  $ r&   r|   )rE   ry  s    €r4   Úignore_overloadedÚ3infer_methods_to_compile.<locals>.ignore_overloadedo  s   ø€ ØÐ"8Ñ8Ð8r6   )r(  rc   rÁ   r<   r6  rŽ   rB   r(   r�   r   rH   rI   rJ   rK   rL   r   rÅ   rÃ   rÄ   r‹  r£   Úfilterr]   r^   rF   )rD   rÚ   ÚmethodsÚforward_funcÚmodule_forwardÚexportedr?   rO   rx  r…  r“  Úfiltered_methodsÚuniquerÚuniqued_methodsrl  rN   ry  s                   @r4   Úinfer_methods_to_compiler�  I  sž  ø€ ô
 ˜YÔ'Ü/°	Ó:Ðà€GÜˆy˜)×$Ñ$¬]×-HÒ-HØ×Ñ÷.ñ .ô ˜y×0Ñ0°*¸dÓCˆÜ ¤§¡§¡°)¸TÓBˆØÓ)Ø �kˆGà€HÜ�I–ˆØÓ%ÙÜ�y¨Ó-ˆä×2Ò2°4Ó8Ü×.Ñ.×5Ñ5ô6ð �O‰O˜DÖ!ñ ð Ñ €Gä!¤'¨)°_ÀbÓ"IÓJÐÜ,¨YÓK€MØ×!Ñ!Ô";¸MÓ"JÔKÜ-¨mÓ<€Nà4€IÔõ9ô Ð/Ó9Ðô ›€GØ€OÛ ˆØ‹?ÙØ×Ñ˜tÔ$Ø�‰�DÖñ	 !ñ ETÓTÂO¸&Ô" 9Ö5ÁO€EÐTØÑ!Ð!ùò Us   Æ#F>c           	      ó  • [        U 5        0 n/ nU R                  R                  5        H™  nUR                  U;   aT  [	        U5      [	        XR                     5      :w  a/  [        SUR                   S[        U 5      R                   S35      eOX1UR                  '   UR                  [        X3R                  5      5        M›     / nU R                  R                  5        H™  nUR                  U;   aT  [	        U5      [	        XR                     5      :w  a/  [        SUR                   S[        U 5      R                   S35      eOXQUR                  '   UR                  [        XUR                  5      5        M›     X$4$ )z3Return forward hook and pre_hook ScriptModuleStubs.zHook 'z' on zV has at least two different python definitions. Please use unique names for all hooks.z
Pre-hook ')
r(  r   rÕ   r'   Úidr•   r\   rL   r@   r   )rD   Úhook_mapr  rê   r  rë   s         r4   rH  rH  ‚  sg  € ä˜YÔ'Ø€Hà€JØ×(Ñ(×/Ñ/Ö1ˆØ�=‰=˜HÓ$Ü�$‹xœ2˜h§}¡}Ñ5Ó6Ó6Ü"Ø˜TŸ]™]˜O¨5´°i³×1IÑ1IÐ0Jð K>ð >óð ð 7ð '+�T—]‘]Ñ#Ø×Ñœ) D¯-©-Ó8Ö9ñ 2ð €NØ×0Ñ0×7Ñ7Ö9ˆØ×Ñ Ó(Ü�(‹|œr (×+<Ñ+<Ñ"=Ó>Ó>Ü"Ø  ×!2Ñ!2Ð 3°5¼¸i»×9QÑ9QÐ8Rð S>ð >óð ð ?ð +3�X×&Ñ&Ñ'Ø×Ñœi¨×2CÑ2CÓDÖEñ :ð Ð%Ð%r6   c                 ó²  • [        U 5      n[        USS9n0 n[        U5       Hu  n[        XS5      n[	        U[
        5      (       d  M&  UR                  (       d  [        SU SU R                   S35      e[        R                  " UR                  5      X4'   Mw     U Vs/ s H)  n[        X6R                  5       R                     U5      PM+     nnU$ s  snf )zjCreate property stubs for the properties of the module by creating method stubs for the getter and setter.r9   r:   Nz	Property z of z must have a getter)r\   r	   rH   rB   rC   rV   rb  r•   r'   r<   r=   r   r?   )rD   Ú	module_tyÚproperties_astsÚrcbsr?   rO   r2   rN   s           r4   rG  rG  ¤  sÁ   € ä�Y“€IÜ*¨9Ð@WÑX€OØ€Dä�I–ˆÜ�y¨Ó-ˆÜ�dœH×%Ó%Ø—9—9Ü"Ø ˜v T¨)×*<Ñ*<Ð)=Ð=PÐQóð ô '×JÒJÈ4Ï9É9ÓUˆD‹Jñ ñ BQÓQÂ¸#Œ\˜$Ÿx™x›zŸ™Ñ/°Ö5Á€EÐQØ€Lùò Rs   Â 0Cc                 ó’   ^ • [        U[        R                  R                  5      (       a  U$ [	        U5        U 4S jn[        X5      $ )a  
Make a ScriptModule from an nn.Module, using the interface methods rule for determining which methods to compile.

Args:
    mod_interface: the interface type that the module have
    nn_module:  The original Python nn.Module that we are creating a ScriptModule for.
c                 ód   >• TR                  5        Vs/ s H  n[        X5      PM     nnU$ s  snf )z…Rule to infer the methods from the interface type.

It is used to know which methods need to act as starting points for compilation.
)ÚgetMethodNamesrF   )rD   rl  rN   Úmod_interfaces      €r4   Ú"infer_interface_methods_to_compileÚ<interface_script.<locals>.infer_interface_methods_to_compileÅ  s?   ø€ ð (×6Ñ6Ô8ó
â8�ô " )Ö4Ù8ð 	ð 
ð ˆùò	
s   ”-)rC   r(   r)   r  r(  r-  )r¨  rD   r©  s   `  r4   r4  r4  ¸  s<   ø€ ô �)œUŸY™Y×3Ñ3×4Ñ4ØÐä˜YÔ'õ	ô   	ÓNÐNr6   c                 óÊ  • [         R                  " U 5      (       a  g [        U [        R                  R
                  5      (       a  g [        R                  " U 5      (       d-  [        R                  " U 5      (       d  [        SU  SU  S35      e[        U S5      (       a  U R                  5       OU n [         R                  " U 5      n[        R                  R                  XS9$ )NÚ`z|` is not a function. Recursive scripting only supports Python functions or methods currently.
Consider manually annotating `z` with @torch.jit.script.Ú__prepare_scriptable__)rŠ  )r<   r6  rC   r(   r�   r   rƒ   rÍ   r5  r•   rÁ   r­  r=   r)   rÎ   )r\  Úlocr/   s      r4   Útry_compile_fnr¯  Ó  sÂ   € Ü×"Ò" 2×&Ñ&àä�"”e—h‘h—o‘o×&Ñ&ð ä×Ò˜b×!Ñ!¬'×*:Ò*:¸2×*>Ñ*>ÜØ�ˆtð -à-/¨DÐ0IðKó
ð 	
ô )0°Ð4L×(MÑ(Mˆ×	"Ñ	"Ô	$ÐSU€Bô
 ×
;Ò
;¸BÓ
?€CÜ�9‰9×Ñ˜BÐÐ)Ð)r6   c                 ó@   • [         R                  R                  U 5      $ )z;Wrap this torch._C.Object in a Python RecursiveScriptClass.)r(   r)   ÚRecursiveScriptClass)Ú	cpp_classs    r4   r"  r"  ï  s   € ä�9‰9×)Ñ)¨)Ó4Ð4r6   c                 ó^   • SS jn[         R                  R                  R                  X5      $ )zYWrap this torch._C.ScriptModule in a Python ScriptModule, recursively for all submodules.c                 ó"  • [         R                  R                  U R                  5      R	                  5        H  u  p[        X[        U5      5        M     [         R                  R                  R                  U R                  R                  5       5      U l
        [        U R                  R                  5       5       H  u  p4X@R                  U'   M     [        U R                  R                  5       5       H  u  p4X@R                  U'   M     g r&   )r(   r,   r¨   rL  rU   r!  Úwrap_cpp_modulerº   r»   Ú_typer  rK  rM  r   rN  r   )r:  r?   r@  r[  r\  s        r4   rA  Ú wrap_cpp_module.<locals>.init_fn÷  sÌ   € Ü %§¡× 3Ñ 3°M×4DÑ4DÓ E× KÑ KÖ MÑˆDÜ�M¬¸Ó)DÖEñ !Nä',§x¡x×'BÑ'B×'PÑ'PØ×Ñ×"Ñ"Ó$ó(
ˆÔ$ô ! ×!1Ñ!1×!HÑ!HÓ!JÖK‰GˆCØ46×,Ñ,¨SÓ1ñ Lä  ×!1Ñ!1×!DÑ!DÓ!FÖG‰GˆCØ02×(Ñ(¨Ó-ò Hr6   rþ   )r(   r)   r9   rI  )r@  rA  s     r4   rµ  rµ  ô  s$   € ô
3ô �9‰9×*Ñ*×5Ñ5°jÓJÐJr6   c                 óô   • [         R                  " U5      (       a  g [        XR                  5      n[        R                   R                  5          [        X4S5        S S S 5        U$ ! , (       d  f       U$ = f)Nr|   )r<   r6  r@   r'   r(   Ú_disable_emit_hooksr  )rû   r\  Ústubs      r4   Úcompile_unbound_methodr»    s`   € Ü×"Ò" 2×&Ñ&ØÜ�RŸ™Ó%€DÜ	×	Ñ	×	0Ñ	0Õ	2ô 	1°ÀÈÔL÷ 
3ð €K÷	 
3Ô	2ð €Kús   ÁA(Á(
A7c                 óˆ   ^ ^• U U4S jnTUl         TR                  Ul        [        R                  R	                  TU5        U$ )zÕ
Return a function that lazily binds `unbound_method` to a provided Module IValue, then invokes the method.

We do this so that any Python shenanigans that
will poison type sharing are impossible at compile time.
c                 óž   >• SU4S jjn[         R                  R                  R                  X5      n[        R
                  " TU5      nU" U6 $ )Nc                 ó  >• TR                   n[        U5       H8  n[        XS 5      n[        R                  " U5      (       d  M,  [        XU5        M:     TR                  5       R                  5        H  u  p$[        XU5        M     g r&   )Úpy_classrH   rB   r<   r6  r!  Úget_constantsrU   )r:  Ú
orig_classr?   rO   rá   rû   s        €r4   rA  Ú7lazy_bind.<locals>.lazy_binding_method.<locals>.init_fn  sp   ø€ Ø&×/Ñ/ˆJô ˜Jž�Ü˜z°Ó6�Ü ×.Ò.¨t×4Ó4Ü˜M°Ö6ñ (ð  -×:Ñ:Ó<×BÑBÖD‘�Ü˜¨UÖ3ò  Er6   rþ   )r(   r)   r9   rI  ÚtypesÚ
MethodType)r@  ÚargsrA  r:  rl  rû   Úunbound_methods        €€r4   Úlazy_binding_methodÚ&lazy_bind.<locals>.lazy_binding_method  s@   ø€ ÷	4ô Ÿ	™	×7Ñ7×BÑBÀ:ÓWˆÜ×!Ò! .°-Ó@ˆÙ�tˆ}Ðr6   )Úoriginal_fnr'   r(   r<   Úcopy_torchscript_modifier)rû   rÆ  rÇ  s   `` r4   Ú	lazy_bindrË    s?   ù€ öð( '5ÐÔ#Ø#1×#:Ñ#:ÐÔ Ü	×Ñ×1Ñ1°.ÐBUÔVàÐr6   )Trþ   )TF)LÚcollectionsrU  rƒ   rv   rÃ  r¿   r(   Útorch._jit_internalr<   Útorch._sourcesr   Útorch.jit._builtinsr   Útorch.jit._checkr   Útorch.jit._stater   r   r   Útorch.jit.frontendr	   r
   r   r   Útorch.nnr   Ú
namedtupler   r   rÈ   r5   r@   rF   rP   rc   ÚboolÚfloatÚintrr  r\   ÚdeviceÚlayoutÚdtypeÚqschemerq   rf   r,   Ú_jit_tree_viewsÚSourceRangeFactoryrz   r„   rì   rî   r  r  r  r¼   r&  r-  r*  rh  rZ  rÅ   rÄ   r�  r‹  r(  r�  rH  rG  r4  r¯  r"  rµ  r»  rË  r|   r6   r4   Ú<module>rÞ     s„  ðã Û Û Û Û Û ã Ý +Ý %Ý -Ý <ß MÑ M÷ó õ ð ×)Ò)ØÐJóÐ ð ×%Ò% nÐ6UÓV€òÐ ò,ò,ò(ò
ò+ð, 	Ø	ØØÙˆƒJØ	‡L�LØ	‡L�LØ	‡K�KØ	‡M�Mð
€òô&	�E—H‘H×,Ñ,×?Ñ?ô 	ò$ô.v!÷rñ ñ< (Ó)Ð ðà	ôôTôòD&ô,Iò2Xò|'ô9ò
òB"ô
òô&ò,6"òr&òDò(Oò6*ò85ò
Kò$ó r6   