Ë
    f^(hp¥  ã                   óŠ  — d dl Z d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	 d dl
mZ d dlmZ d dlmZ d dlmZmZmZ d dlmZmZmZmZ d dlmZ  e j6                  dd	«      Z e j6                  d
d«      Zg d¢Zd„ Zd„ Z d„ Z!d„ Z"d„ Z#e$e%e&e' e(d«      ejR                  ejT                  ejV                  fZ,d„ Z- G d„ dej\                  j^                  j`                  «      Z1d„ Z2d/d„Z3 G d„ d«      Z4 e4«       Z5d„ Z6d„ Z7d/d„Z8d„ Z9d0d„Z:d„ Z;d„ Z<d „ Z=d!„ Z>d"„ Z?d#„ Z@d$„ ZAd%„ ZBd&„ ZCd'„ ZDd(„ ZEd)„ ZFd*„ ZGd+„ ZHd,„ ZId-„ ZJd.„ ZKy)1é    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                 óú   — t        | «      }|smt        | | j                  «      }t        j                  j
                  j                  | «      }t        j                  j                  ||||«      }t        | |«       |S ©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         úR/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/torch/jit/_recursive.pyÚ_compile_and_register_classr5   6   sh   € Ü$ SÓ)€LáÜ  S§\¡\Ó2ˆÜ—9‘9×%Ñ%×@Ñ@ÀÓEˆÜ—x‘x×9Ñ9Ø˜C ¨3ó
ˆô 	˜#˜|Ô,àÐó    c                 ób   — t        j                  | «      }t        | |d¬«      }t        ||| «      S ©NÚRecursiveScriptModule©Ú	self_name)Ú_jit_internalÚ#createResolutionCallbackFromClosurer   r   )ÚfuncÚnamer/   r2   s       r4   Ú	make_stubr@   D   s0   € Ü
×
;Ñ
;¸DÓ
A€CÜ
�d˜DÐ,CÔ
D€CÜ˜C  dÓ+Ð+r6   c                 óV   — t        | |«      }t        |t        «      r|S t        ||«      S r&   )ÚgetattrÚ
isinstancer   r@   )Ú	nn_moduleÚmethod_namer>   s      r4   Úmake_stub_from_methodrF   J   s-   € Ü�9˜kÓ*€DÜ�$Ô(Ô)Øˆô �T˜;Ó'Ð'r6   c                 óÚ   — g }t        | «      D ]Z  }t        | |d «      }t        j                  |«      t        j                  j
                  u sŒ@|j                  t        | |«      «       Œ\ |S r&   )ÚdirrB   r<   Úget_torchscript_modifierÚFunctionModifiersÚEXPORTÚappendrF   )ÚmodÚstubsr?   Úitems       r4   Ú make_stubs_from_exported_methodsrP   Y   sg   € Ø€EÜ�C“ò ;ˆÜ�s˜D $Ó'ˆä×2Ñ2°4Ó8Ü×.Ñ.×5Ñ5ò6ð �L‰LÔ.¨s°DÓ9Õ:ð;ð €Lr6   c                 ó–   — t        | dg «      }d„ } |t        | «      «      }t        «       }|D ]  }||v sŒ|j                  |«       Œ |S )NÚ__jit_ignored_attributes__c                 óˆ   — t        | «      j                  «       D ��ch c]  \  }}t        |t        «      sŒ|’Œ c}}S c c}}w r&   )ÚvarsÚitemsrC   Úproperty)ÚmoduleÚkÚvs      r4   Úget_properties_namesz4jit_ignored_properties.<locals>.get_properties_namesk   s0   € Ü" 6›l×0Ñ0Ó2×N‘d�a˜´jÀÄHÕ6M’ÓNÐNùÓNs   �>¶>)rB   ÚtypeÚsetÚadd)rW   Ú!user_annotated_ignored_attributesrZ   Ú
propertiesÚuser_annoted_ignored_propertiesÚignored_attrs         r4   Újit_ignored_propertiesrb   f   s`   € Ü(/ØÐ,¨bó)Ð%òOñ &¤d¨6£lÓ3€JÜ&)£eÐ#à9ò >ˆØ˜:Ò%Ø+×/Ñ/°Õ=ð>ð +Ð*r6   c                 óF  ‡ ‡— t        |t        «      r|S t        |t        t        f«      rt        ˆ ˆfd„|D «       «      S dj	                  d„ t        D «       «      }t        t        j                  dt        j                  t        |«      «      › d‰› d‰ › d|› d�	«      «      ‚)	Nc              3   ó8   •K  — | ]  }t        ‰|‰«      –— Œ y ­wr&   )Ú_get_valid_constant)Ú.0ÚxÚattrÚ
owner_types     €€r4   ú	<genexpr>z&_get_valid_constant.<locals>.<genexpr>‹   s   øè ø€ ÒIÀ!Ô(¨¨q°*×=ÑIùs   ƒz, c              3   óF   K  — | ]  }t        j                  |«      –— Œ y ­wr&   )r(   Útypename)rf   Útyps     r4   rj   z&_get_valid_constant.<locals>.<genexpr>Œ   s   è ø€ ÒI°#œ%Ÿ.™.¨×-Ñ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(   rl   r[   )rh   rY   ri   Ú	constantss   ` ` r4   re   re   ‡   sš   ù€ Ü�!”_Ô%ØˆÜ	�Aœœt�}Ô	%ÜÔIÀqÔIÓIÐIØ—	‘	ÑI¼ÔIÓI€IÜ
Ü�‰ð
Ü
�.‰.œ˜a›Ó
!Ð	"Ð"9¸*¸ÀQÀtÀfð Mð (˜[ð )	ð	ó	
ó
ð 
r6   c                   ó   ‡ — e Zd Zˆ fd„Zˆ xZS )ÚSourceContextc                 ó*   •— t         ‰| �  ||||«       y r&   )ÚsuperÚ__init__)ÚselfÚsourceÚfilenameÚfile_linenoÚleading_whitespace_lenÚ	__class__s        €r4   r{   zSourceContext.__init__›   s   ø€ Ü‰Ñ˜ ¨;Ð8NÕOr6   )r'   Ú
__module__Ú__qualname__r{   Ú__classcell__)r�   s   @r4   rx   rx   š   s   ø„ ÷Pð Pr6   rx   c                 óÊ   ‡— t         j                  dk  rt        | di «      S t        j                  | «      }|r|S ˆfd„Št        | t        «      r| n
t        | «      } ‰|«      S )N)é   é
   Ú__annotations__c                 óx   •— t        j                  | «      }|r|S | j                  D ]  } ‰|«      }|sŒ|c S  i S r&   )ÚinspectÚget_annotationsÚ	__bases__)ÚclsÚcls_annotationsÚbaseÚget_cls_annotationss      €r4   r�   z,get_annotations.<locals>.get_cls_annotationsª   sI   ø€ Ü!×1Ñ1°#Ó6ˆÙØ"Ð"Ø—M‘Mò 	'ˆDÙ1°$Ó7ˆOÚØ&Ò&ð	'ð ˆ	r6   )ÚsysÚversion_inforB   rŠ   r‹   rC   r[   )r.   Úannotationsr�   r�   s      @r4   r‹   r‹   Ÿ   sb   ø€ Ü
×Ñ˜'Ò!Ü�sÐ-¨rÓ2Ð2ô
 ×)Ñ)¨#Ó.€KÙØÐôô ˜C¤Ô&‰#¬D°«I€CÙ˜sÓ#Ð#r6   c                 óJ  ‡— t         j                  j                  t        | «      «      }t	        | t         j
                  j                  «      r|j                  «        t	        | t         j
                  j                  t         j
                  j                  f«      r|j                  «        t	        | t         j
                  j                  «      r|j                  «        t	        | t         j
                  j                  «      r|j                  «        t        | «      Št	        | t         j                   j"                  j$                  «      ri Št'        | dg «      }|j)                  |«       t+        | «      }ˆfd„}t-        «       }| j.                  j1                  «       D ]g  \  }}||v rŒ|�t	        |t         j2                  «      sJ ‚ |||«      \  }	}
|j5                  ||	j                  «       dd«       |j7                  |«       Œi | j8                  j1                  «       D ]g  \  }}||v rŒ|�t	        |t         j2                  «      sJ ‚ |||«      \  }	}
|j5                  ||	j                  «       dd«       |j7                  |«       Œi | j:                  j1                  «       D ]Ò  \  }}||v rŒ |||«      \  }	}
|€#|j5                  ||	j                  «       dd«       Œ<|	j=                  «       rX|	j                  «       j?                  «       sJ ‚t         j                  j@                  jC                  |	j                  «       «      }ntE        ||«      }|jG                  ||«       |j7                  |«       ŒÔ t-        t'        | dd«      «      }‰j1                  «       D ]6  \  }}t         jH                  jK                  |«      sŒ&|j7                  |«       Œ8 |D ]Ô  }||v r[|| j:                  v rd}n-|| j8                  v rd	}n|| j.                  v rd
}ntM        d«      ‚tO        jP                  d|› d|› d�«       ŒbtS        | |«      stO        jP                  d|› d�«       Œˆt'        | |«      }|jU                  |tW        ||t        | «      jX                  «      «       |j7                  |«       ŒÖ t'        | di «      }|j[                  t]        t_        | |«      «      «       |j1                  «       D ]  \  }}|ja                  ||«       Œ | jb                  j1                  «       D �]æ  \  }}|td        v s|jg                  d«      rŒ!||v rŒ&||v rŒ+t	        |t         jh                  jj                  «      }|r|jl                  }to        jp                  |«      r_	 t         jr                  ju                  |«      }|jw                  |t         j                  jy                  |«      j                  «       |«       ŒÑt        |«      }|r|j�                  ||«       Œñt	        |t         jr                  j‚                  «      r@|jw                  |t         j                  jy                  |«      j                  «       |«       �ŒU |||«      \  }	}|	j=                  «       r$|j5                  ||	j                  «       dd«       �Œ•|rdnd}|	j…                  «       › d|› �}dt        j†                  t        |«      «      › d|› d�}|j}                  ||«       �Œé | jˆ                  j‹                  «       D ]  }|j�                  |«       Œ | jŽ                  j‹                  «       D ]  }|j‘                  |«       Œ |S # tz        $ r"}d|› �}|j}                  ||«       Y d}~�Œ¢d}~ww xY w)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           	      ó¬  •— d}	 | ‰v r‚‰|    t         j                  j                  j                  d   k7  rUt         j                  j
                  j                  ‰|    t        «       «      }t         j                  j                  |«      }n¡t        |t         j                  j                  «      r\t         j                  j
                  j                  |j                  t        «       «      }t         j                  j                  |«      }n!t         j                  j                  |«      }d}||fS # t        $ r}t        d| › d|› d|› �«      |‚d }~ww xY w)NFÚforwardTzError inferring type for z: )r(   Únnr   rˆ   r)   r“   Úann_to_typer   r,   ÚInferredTyperC   Ú	Attributer[   Ú_jit_try_infer_typeÚRuntimeError)r?   rO   Úinferredr˜   Ú	attr_typeÚreÚclass_annotationss         €r4   Ú
infer_typez/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Ü#Ÿi™i×3Ñ3×?Ñ?ÀÇ	Á	Ì:Ë<ÓX�Ü!ŸH™H×1Ñ1°+Ó>‘	ä!ŸH™H×8Ñ8¸Ó>�	Ø�ð ˜(Ð"Ð"øô ò 	YÜÐ!:¸4¸&ÀÀ4À&ÈÈ2È$ÐOÓPÐVXÐXûð	Yús   …D'D0 Ä0	EÄ9EÅENTFÚ__constants__© Ú	submoduleÚbufferÚ	parameterz3added_names must be submodule, parameter, or bufferú'zA' was found in ScriptModule constants,  but it is a non-constant z. Consider removing it.zb' 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_attributesrb   r\   r   rU   ÚTensorÚadd_attributer]   r   r"   ÚsuccessÚis_interface_typeÚConcreteModuleTypeÚfrom_jit_typeÚget_module_concrete_typeÚ
add_moduler<   Úis_finalÚAssertionErrorÚwarningsÚwarnÚhasattrÚadd_constantre   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Úreasonrl   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ñ   ¸   s8  ø€ ô "ŸH™H×>Ñ>¼tÀI»ÓOÐÜ�)œeŸh™h×1Ñ1Ô3Ø×-Ñ-Ô/Ü�)œeŸh™h×1Ñ1´5·8±8×3FÑ3FÐGÔHØ×-Ñ-Ô/Ü�)œeŸh™h×4Ñ4Ô6Ø×0Ñ0Ô2Ü�)œeŸh™h×4Ñ4Ô6Ø×0Ñ0Ô2ä'¨	Ó2ÐÜ�)œeŸh™h×3Ñ3×@Ñ@ÔBØÐô )0ØÐ/°ó)Ð%ð ×0Ñ0Ð1RÔSÜ/°	Ó:Ðô#ô8 “%€Kà×+Ñ+×1Ñ1Ó3ò ‰
ˆˆdØÐ4Ñ4Øàˆ|œz¨$´·±Ô=Ð=Ð=Ù! $¨Ó-‰ˆ	�1ð 	×+Ñ+¨D°)·.±.Ó2BÀDÈ%ÔPØ�‰˜Õðð  ×(Ñ(×.Ñ.Ó0ò ‰
ˆˆdØÐ4Ñ4Øàˆ|œz¨$´·±Ô=Ð=Ð=Ù! $¨Ó-‰ˆ	�1Ø×+Ñ+¨D°)·.±.Ó2BÀEÈ4ÔPØ�‰˜Õðð  ×(Ñ(×.Ñ.Ó0ò ‰
ˆˆdØÐ4Ñ4Øá! $¨Ó-‰ˆ	�1Øˆ<ð "×/Ñ/°°i·n±nÓ6FÈÈuÔUØØ×ÑÔØ—>‘>Ó#×5Ñ5Ô7Ð7Ð7ä %§¡× ;Ñ ;× IÑ IØ—‘Ó ó!Ñô
 !9¸¸{Ó KÐØ×(Ñ(¨Ð/@ÔAà�‰˜Õð+ô0 œ 	¨?¸BÓ?Ó@€Mð '×,Ñ,Ó.ò $‰	ˆˆcÜ×Ñ×'Ñ'¨Õ,Ø×Ñ˜dÕ#ð$ð ò !ˆØ�;Ñð �y×)Ñ)Ñ)Ø"‘Ø˜×+Ñ+Ñ+Ø‘Ø˜×.Ñ.Ñ.Ø"‘ä$ØIóð ô �M‰MØ�D�6ð -Ø-1¨FÐ2IðKôð Ü�y $Ô'ô �M‰MØ�D�6ð (ð (ôð
 Ü˜	 4Ó(ˆØ×*Ñ*ØÔ% d¨E´4¸	³?×3KÑ3KÓLô	
ð 	�‰˜ÕðC!ôH ˜	 ?°BÓ7€Ià×ÑÜ!Ü$ YÐ0BÓCó	
ôð
 #,§/¡/Ó"3ò CÑˆÐØ×*Ñ*¨4Ð1AÕBðCð !×)Ñ)×/Ñ/Ó1ó DC‰ˆˆeØÔ%Ñ%¨¯©¸Ô)>ð àÐ4Ñ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Ø!×8Ñ8Ø”e—h‘h×2Ñ2°5Ó9×>Ñ>Ó@À%ôñ ñ )¨¨uÓ5Ñˆ	�8Ø×ÑÔØ!×/Ñ/°°i·n±nÓ6FÈÈuÖUñ ñ Yàð ð
 "+×!1Ñ!1Ó!3Ð 4°B°|°nÐEˆOð:Ü:?¿.¹.ÌÈeËÓ:UÐ9Vð W*Ø*9Ð):¸!ð=ð ð
 "×6Ñ6°t¸TÖBðIDCðN ×(Ñ(×/Ñ/Ó1ò 5ˆØ×.Ñ.¨tÕ4ð5à×0Ñ0×7Ñ7Ó9ò =ˆØ×2Ñ2°8Õ<ð=ð !Ð øôk ò 	Gð
>à>?¸SðBð ð
 &×:Ñ:¸4À×FÒFûð	Gús   ÖA]7Ý7	^"Þ ^Þ^"c                   óœ   — e Zd ZU eee   eej                  j                     f   e
d<   eej                  j                     e
d<   dd„Zd„ Zy)ÚConcreteTypeStoreÚ
type_storeÚmethods_compiledNc                 ó0   — i | _         t        «       | _        y r&   )rô   r\   rõ   )r|   s    r4   r{   zConcreteTypeStore.__init__ª  s   € àˆŒä #£ˆÕr6   c                 ó  — t        |«      }t        |«      }|| j                  vrg | j                  |<   | j                  |   }|D ]  }|j                  |«      sŒ|c S  |j	                  «       }| j                  |   j                  |«       |S )zbInfer a ConcreteType from this `nn.Module` instance. Underlying JIT types are re-used if possible.)rñ   r[   rô   ÚequalsÚbuildrL   )r|   rD   rÞ   Únn_module_typeÚknown_typesÚ
known_typeÚconcrete_types          r4   Úget_or_create_concrete_typez-ConcreteTypeStore.get_or_create_concrete_type°  s‘   € ä ;¸IÓ FÐä˜i›ˆØ §¡Ñ0Ø.0ˆD�O‰O˜NÑ+ð —o‘o nÑ5ˆØ%ò 	"ˆJØ× Ñ Ð!6Õ7Ø!Ò!ð	"ð
 .×3Ñ3Ó5ˆØ�‰˜Ñ'×.Ñ.¨}Ô=ØÐr6   )ÚreturnN)r'   r‚   rƒ   Údictr[   r   rq   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                 ól  — |D �cg c]  }|j                   ‘Œ }}|D �cg c]  }|j                  ‘Œ }}|D �cg c]  }t        |j                  «      ‘Œ }}|D �cg c]  }|j                   ‘Œ }}|D �cg c]  }|j                  ‘Œ }	}| j	                  ||	|||«       y c c}w c c}w c c}w c c}w c c}w 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�1—6“6Ð0€KÐ0Ø2>Ö?¨Q�1×(Ó(Ð?€KÐ?ØDPÖQ¸qÔ'¨×(9Ñ(9Õ:ÐQ€OÐQà%3Ö4 �Q—V“VÐ4€MÐ4Ø4BÖC¨q�Q×*Ó*ÐC€MÐCà×0Ñ0Ø�} k°;Àõùò 1ùÚ?ùÚQùâ4ùÚCs   …BžB"·B'ÁB,Á2B1c                 ó  — |D �cg c]  }|j                   ‘Œ }}|D �cg c]  }|j                  ‘Œ }}|D �cg c]  }|j                   ‘Œ }}|D �cg c]  }|j                  ‘Œ }}| j                  ||||«       y c c}w c c}w c c}w c c}w 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 �Q—V“VÐ4€MÐ4Ø4BÖC¨q�Q×*Ó*ÐC€MÐCà×Ñ 	¨9°mÀ]ÕSùò -ùÚ;ùâ4ùÚCs   …A:žA?·BÁB	c                 ó,  — t        | t        «      sJ ‚t        | t        j                  j                  «      rt        | d«      r| j                  S |rt        j                  | «      }|S t        | |«      }|j                  «        |j                  «       }|S )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.
    Ú_concrete_type)rC   r   r(   r)   ÚScriptModulerÆ   r  Úconcrete_type_storerþ   rñ   Úset_poisonedrù   )rD   rÝ   rý   rÞ   s       r4   rÀ   rÀ   à  s�   € ô �i¤Ô(Ð(Ð(Ü�)œUŸY™Y×3Ñ3Ô4¼ØÐ#ô:ð ×'Ñ'Ð'áä+×GÑGÈ	ÓRˆð Ðô	 !<¸IÀ{Ó SÐØ×*Ñ*Ô,Ø-×3Ñ3Ó5ˆàÐr6   c                 óŒ  — t        j                  t        | «      «      }t        j                  t        | «      «      }t	        t        | «      ||«       t        j                  |«      }t        j                  j                  |«      }| j                  j                  «       D ]  \  }}|j                  ||«       Œ t        |«      S )z
    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Ð0DÔEÜ×#Ñ#Ð$8Ó9€Hä—‘×2Ñ2°8Ó<€Jà—|‘|×)Ñ)Ó+ò (‰ˆˆeØ×Ñ˜4 Õ'ð(ô ˜*Ó%Ð%r6   c                 óÌ   — t        | t        j                  j                  «      rJ ‚t	        | «       t        | |«      }|st        «       j                  | «       t        | ||«      S )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.
    )	rC   r(   r)   r9   Úcheck_module_initializedrÀ   r   ÚcheckÚcreate_script_module_impl)rD   Ústubs_fnrÝ   Ú
is_tracingrý   s        r4   Úcreate_script_moduler-    sU   € ô  ˜)¤U§Y¡Y×%DÑ%DÔEÐEÐEÜ˜YÔ'Ü,¨Y¸ÓD€MÙÜ'Ó)×/Ñ/°	Ô:Ü$ Y°¸xÓHÐHr6   c                 óh  ‡ ‡‡‡‡— t         j                  j                  ‰j                  «      Š ‰‰ «      }t	        ‰ «      }t        ‰ «      \  }}t        ‰ «      Šˆˆˆˆ ˆfd„}t         j                  j                  j                  ‰|«      }‰t        j                  vrXt        ‰||«       t        ‰||«       t         j                  j                  ‰«       t        j                  j                  ‰«       t!        |j"                  j%                  «       «      D ]  \  }	}
|
|j&                  |	<   Œ t!        |j"                  j)                  «       «      D ]  \  }	}
|
|j*                  |	<   Œ t-        ‰ t         j.                  j0                  t         j.                  j2                  t         j.                  j4                  f«      r0d‰j7                  «       vr|j9                  dt;        ‰ «      › d�«       t-        ‰ t         j.                  j4                  «      rtd‰j7                  «       vrbt;        ‰ j=                  «       «      r8t?        tA        ‰ j=                  «       «      «      }|j9                  d|› d�«       n|j9                  d«       |D ]’  }|jB                  €Œ|jB                  jD                  }||jF                  jI                  «       jH                  k7  rŒN‰jK                  |«      } tM        jN                  |jB                  «      |«      }||jP                  |<   Œ” |D ]¬  }|jF                  jI                  «       jH                  }‰jK                  |jF                  jS                  «       jH                  «      }|jF                  jU                  «       }|r‰jK                  |jH                  «      nd}tW        |||«      |jP                  |<   Œ® tY        ‰ «      D ]Q  }|‰v rŒt[        ‰ |d«      }t]        j^                  |«      t\        j`                  jb                  u sŒEte        |‰ |«       ŒS |S )	aR  
    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.
    c                 óì  •— ‰	j                  «       j                  «       D ]R  }t        ‰|«      }t        |t        j
                  j                  «      r|j                  n|}‰
j                  ||«       ŒT ‰	j                  «       D ]Æ  \  }}t        ‰|«      }t        |t        «      sJ dt        |«      › �«       ‚|j                  }t        |t        j                  j                  «      rt        ||«      }n4t        |t        j
                  j                   «      r|}nt#        ||‰«      }‰
j                  ||«       || j$                  |<   ŒÈ t'        ‰«      D ]’  }|‰v rŒt        ‰|d «      }t)        j*                  |«      rJt-        j.                  |«      r5t        ‰|«      j0                  }|j3                  | «      }t        | ||«       Œt‰	j5                  |«      sŒ†t        | ||«       Œ” ‰	| _        y )NzExpected Module but got )Úget_attributesÚkeysrB   rC   r(   r)   rš   ræ   r!  Úget_modulesr   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_fnz*create_script_module_impl.<locals>.init_fn>  sÞ  ø€ ð "×0Ñ0Ó2×7Ñ7Ó9ò 	1ˆDÜ  ¨DÓ1ˆJô ˜j¬%¯)©)×*=Ñ*=Ô>ð × Ò àð ð
 ×Ñ˜t ZÕ0ð	1ð (5×'@Ñ'@Ó'Bò 	4Ñ#ˆDÐ#Ü  ¨DÓ1ˆJÜØœFôð =à)¬$¨zÓ*:Ð);Ð<ó=ð ð ,×4Ñ4ˆKÜ˜+¤u§x¡x×'=Ñ'=Ô>ä+¨K¸ÓD‘Ü˜J¬¯	©	×(>Ñ(>Ô?Ø%‘ô 5ØÐ 1°8ó�ð ×Ñ˜t XÔ.Ø+3ˆM×"Ñ" 4Ò(ð%	4ô, ˜	“Nò 		3ˆDØÐ)Ñ)ØÜ˜9 d¨DÓ1ˆDÜ×Ñ Ô%¬-×*EÑ*EÀdÔ*KÜ#*¨9°dÓ#;×#DÑ#DÐ Ø/×7Ñ7¸ÓF�Ü˜ t¨\Õ:Ø×3Ñ3°DÕ9Ü˜ t¨TÕ2ð		3ð (5ˆÕ$r6   Ú__len__zdef __len__(self):
   return ú
Ú__contains__z3def __contains__(self, key: str):
   return key in z2def __contains__(self, key: str):
   return False
N)3r(   r,   Ú_create_module_with_typer3  Úget_property_stubsÚget_hook_stubsrb   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Úlenr1  Úreprrq   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  rB  r;  ÚidxÚfnr1  Úmethod_stubr?   Úscript_methodÚwrapped_script_methodÚproperty_stubÚproperty_nameÚfgetrW  ÚfsetrO   rA  rß   s   ```                   @@r4   r*  r*  /  s¯  ü€ ô —‘×2Ñ2°=×3IÑ3IÓJ€JÙ˜IÓ&€LÜ'¨	Ó2€NÜ!/°	Ó!:Ñ€J�Ü/°	Ó:Ð÷05ð 05ôf —I‘I×3Ñ3×>Ñ>¸zÈ7ÓS€Mð Ô/×@Ñ@Ñ@Ü0Ø˜<¨ô	
ô
 	  ¨z¸>ÔJÜ�‰×&Ñ& zÔ2Ü×,Ñ,×0Ñ0°Ô?ô ˜]×-Ñ-×DÑDÓFÓGò 3‰ˆˆRØ02ˆ×(Ñ(¨Ò-ð3ä˜]×-Ñ-×@Ñ@ÓBÓCò /‰ˆˆRØ,.ˆ×$Ñ$ SÒ)ð/ô
 	ØœŸ™×+Ñ+¬U¯X©X×-@Ñ-@Ä%Ç(Á(×BUÑBUÐVô	
ð ˜Z×5Ñ5Ó7Ñ7à×ÑÐ=¼cÀ)»nÐ=MÈRÐPÔQä�9œeŸh™h×1Ñ1Ô2Ø *×":Ñ":Ó"<Ñ<äˆy�~‰~ÓÔ Üœ˜YŸ^™^Ó-Ó.Ó/ˆDØ× Ñ ØFÀtÀfÈBÐOõð × Ñ Ð!WÔXð $ò =ˆØ×&Ñ&Ð.ð à×*Ñ*×3Ñ3ˆØ�;×#Ñ#×(Ñ(Ó*×/Ñ/Ò/ð Ø"×.Ñ.¨tÓ4ˆð
 !M¤	§¡°×0KÑ0KÓ LØó!
Ðð (=ˆ×Ñ˜tÒ$ð/=ð4 (ò TˆØ%×*Ñ*×/Ñ/Ó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ˆ×Ñ˜}Ò-ðTô �I“ò NˆØÐ%Ñ%ØÜ�y $¨Ó-ˆä×2Ñ2°4Ó8Ü×.Ñ.×EÑEòFô .¨m¸YÈÕMðNð Ðr6   c                 ó|   — t        | |d «      }|€yt        t        j                  j                  |d «      }|€y||k7  S )NF)rB   r(   r)   r9   )Úscript_modelrh   Úscript_attrÚdefault_attrs       r4   Úscript_model_defines_attrrg  Î  sE   € Ü˜,¨¨dÓ3€KØÐØÜœ5Ÿ9™9×:Ñ:¸DÀ$ÓG€LØÐØØ˜,Ñ&Ð&r6   c                 óf   — t        ||«      r%t        | |«      rt        | |t        ||«      «       y y y r&   )rÆ   rg  r!  rB   )rd  Úorigrh   s      r4   rY  rY  Ø  s2   € Üˆt�TÔÔ8¸ÀtÔLÜ�˜d¤G¨D°$Ó$7Õ8ð  MÐr6   c                 óú  — i }t        t        | «      «      D ]Ü  }||v rŒt        | |d «      }t        |«      sŒ!t	        |d«      sŒ.|j
                  €Œ;t        j                  || j                  «      }|€Œ^|j                  |v r)t        t        j                  d|j                  «      «      ‚t        t        |«      «      D �cg c]  }|dz   t        |«      z   ‘Œ }}t        t!        ||«      «      ||<   ŒÞ |S c c}w )Nr‚   Úmethodr©   )rH   r[   rB   ÚcallablerÆ   r‚   r<   Ú_get_overloaded_methodsr�   r8  rœ   Ú,get_overload_no_implementation_error_messageÚrangerQ  Ústrrq   Úzip)rM   rb   rç   r?   rO   Úmethod_overloadsÚiÚnamess           r4   rÊ   rÊ   Ý  s   € à€Iä”D˜“I“ò AˆØÐ)Ñ)ØÜ�s˜D $Ó'ˆÜ˜Œ~Øô �4˜Õ&¨4¯?©?Ñ+FÜ,×DÑDØ�c—m‘mó Ðð  Ð'Øà�}‰}Ð 0Ñ0Ü"Ü!×NÑNØ  $§-¡-óóð ô 49¼Ð=MÓ9NÓ3OÖP¨a�T˜D‘[¤3 q£6Ó)ÐPˆEÐPÜ"¤3 uÐ.>Ó#?Ó@ˆI�dŠOð/Að2 Ðùò Qs   ÃC8c                 ó    — i }| j                  «       D ]8  \  }}|j                  }||vrg ||<   |D ]  \  }}||   j                  |«       Œ Œ: |S r&   )rU   r'   rL   )Úoverload_infoÚoverload_name_mappingsÚorig_fnrç   Úoriginal_nameÚoverload_namerá   s          r4   rÉ   rÉ   ý  sy   € ð 46ÐØ+×1Ñ1Ó3ò HÑˆ�Ø×(Ñ(ˆØÐ 6Ñ6Ø46Ð" =Ñ1à )ò 	HÑˆM˜1Ø" =Ñ1×8Ñ8¸ÕGñ	HðHð "Ð!r6   c                 óÜ   — t         j                  j                  j                  | d t	        «       t        j                  | «      «      }|€#t        j                  | «      }t        d|› �«      ‚y )Nz>Must explicitly add type annotations to overloaded functions: )
r(   r)   r“   Úget_signaturer   rŠ   r6  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           	      ó„  — g }| j                  «       D ]ª  \  }}t        ||j                  d¬«      }|D ]ˆ  \  }}t        |«       t        ||j                  d¬«      }t        j
                  j                  |j                  «       ||«      }t        j                  |«      }	|j                  t        |	||«      «       ŒŠ Œ¬ |S r8   )rU   r   r'   r  r(   r,   Ú_replace_overloaded_method_declÚdeclr<   r=   rL   r   )
rv  Úoverload_stubsrx  rç   Úorig_astrz  Úoverload_fnÚover_astÚnew_astÚ_rcbs
             r4   Úmake_stubs_for_overloadsr‰    sË   € Ø€NØ+×1Ñ1Ó3ò PÑˆ�ÜØ�W×%Ñ%Ð1Hô
ˆð +4ò 		PÑ&ˆM˜;Ü Ô,Ü"Ø˜[×1Ñ1Ð=TôˆHô —h‘h×>Ñ>Ø—‘“ ¨=óˆGô !×DÑDÀWÓMˆDØ×!Ñ!Ô"2°4¸À+Ó"NÕOñ		Pð	Pð Ðr6   c           	      óÈ  — t        | t        j                  j                  «      sJ ‚t	        | d«      s+t        dt        j                  t        | «      «      › d�«      ‚t	        | d«      sù| j                  j                  «       D ]_  \  }}|€Œ	t        j                  j                  j                  |«      sŒ3t        dt        j                  t        | «      «      › d|› d�«      ‚ | j                  j                  «       D ]_  \  }}|€Œ	t        j                  j                  j                  |«      sŒ3t        dt        j                  t        | «      «      › d|› d�«      ‚ y y )Nr   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œ   rl   r[   r   rU   r¦   Úis_lazyr   )rM   r?   ÚparamÚbufs       r4   r(  r(  )  s9  € Ü�cœ5Ÿ8™8Ÿ?™?Ô+Ð+Ð+Ü�3˜Ô&ÜØ”—‘œt C›yÓ)Ð*Ð*gÐhó
ð 	
ô
 �3Ð+Ô,ØŸ?™?×0Ñ0Ó2ò 	‰KˆD�%ØÑ ¤U§X¡X×%7Ñ%7×%?Ñ%?ÀÕ%FÜ"ØœŸ™¤t¨C£yÓ1Ð2Ð2QÐRVÐQWÐW~Ðóð ð	ð
 Ÿ™×+Ñ+Ó-ò 	‰IˆD�#Ø‰¤5§8¡8×#5Ñ#5×#=Ñ#=¸cÕ#BÜ"ØœŸ™¤t¨C£yÓ1Ð2Ð2NÈtÈfÐT{Ð|óð ñ	ð -r6   c                 ó^  ‡— t        | «       t        | «      }g }t        | d«      rct        j                  | j
                  «      sDt        | j
                  dd«      }t        t        j                  j                  dd«      }||k7  rdg}g }t        | «      D ]U  }||v rŒt        | |d«      }t        j                  |«      t        j                  j                  u sŒE|j                  |«       ŒW ||z   }t        t        | di «      «      Št!        | |«      }‰j#                  t%        |«      «       t'        |«      }	‰| _        ˆfd„}
t+        |
|«      }t-        «       }g }|D ])  }||v rŒ|j                  |«       |j/                  |«       Œ+ |D �cg c]  }t1        | |«      ‘Œ }}|	|z   S c c}w )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–   r8  Nr¨   c                 ó   •— | ‰vS r&   r£   )rE   rw  s    €r4   Úignore_overloadedz3infer_methods_to_compile.<locals>.ignore_overloadedd  s   ø€ ØÐ"8Ð8Ð8r6   )r(  rb   rÆ   r<   r7  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   rv  rƒ  r‘  Úfiltered_methodsÚuniquerÚuniqued_methodsrk  rN   rw  s                   @r4   Úinfer_methods_to_compilerš  >  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Ð8JÓK€MØ×!Ñ!Ô";¸MÓ"JÔKÜ-¨mÓ<€Nà4€IÔô9ô Ð/°Ó9Ðô ›€GØ€OØ ò ˆØ�7‰?ØØ×Ñ˜tÔ$Ø�‰�DÕð	ð ETÖT¸&Ô" 9¨fÕ5ÐT€EÐTØ˜EÑ!Ð!ùò Us   ÆF*c           	      óü  — t        | «       i }g }| j                  j                  «       D ]—  }|j                  |v rSt	        |«      t	        ||j                     «      k7  r>t        d|j                  › dt        | «      j                  › d�«      ‚|||j                  <   |j                  t        ||j                  «      «       Œ™ g }| j                  j                  «       D ]—  }|j                  |v rSt	        |«      t	        ||j                     «      k7  r>t        d|j                  › dt        | «      j                  › d�«      ‚|||j                  <   |j                  t        ||j                  «      «       Œ™ ||fS )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  w  sw  € ä˜YÔ'Ø€Hà€JØ×(Ñ(×/Ñ/Ó1ò 
:ˆØ�=‰=˜HÑ$Ü�$‹xœ2˜h t§}¡}Ñ5Ó6Ò6Ü"Ø˜TŸ]™]˜O¨5´°i³×1IÑ1IÐ0Jð K>ð >óð ð '+ˆH�T—]‘]Ñ#Ø×Ñœ) D¨$¯-©-Ó8Õ9ð
:ð €NØ×0Ñ0×7Ñ7Ó9ò 
FˆØ×Ñ Ñ(Ü�(‹|œr (¨8×+<Ñ+<Ñ"=Ó>Ò>Ü"Ø  ×!2Ñ!2Ð 3°5¼¸i»×9QÑ9QÐ8Rð S>ð >óð ð +3ˆH�X×&Ñ&Ñ'Ø×Ñœi¨°(×2CÑ2CÓDÕEð
Fð �~Ð%Ð%r6   c                 óœ  — t        | «      }t        |d¬«      }i }t        |«      D ]j  }t        ||d«      }t	        |t
        «      sŒ!|j                  st        d|› d| j                  › d�«      ‚t        j                  |j                  «      ||<   Œl |D �cg c])  }t        ||j                  «       j                     |«      ‘Œ+ }}|S c c}w )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   ra  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“ò VˆÜ�y $¨Ó-ˆÜ�dœHÕ%Ø—9’9Ü"Ø ˜v T¨)×*<Ñ*<Ð)=Ð=PÐQóð ô '×JÑJÈ4Ï9É9ÓUˆD�ŠJðVð BQÖQ¸#Œ\˜$˜sŸx™x›zŸ™Ñ/°Õ5ÐQ€EÐQØ€Lùò Rs   Â.C	c                 óˆ   ‡ — t        |t        j                  j                  «      r|S t	        |«       ˆ fd„}t        ||«      S )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                 ó`   •— ‰j                  «       D �cg c]  }t        | |«      ‘Œ }}|S c c}w )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   rk  rN   Úmod_interfaces      €r4   Ú"infer_interface_methods_to_compilez<interface_script.<locals>.infer_interface_methods_to_compileº  s>   ø€ ð (×6Ñ6Ó8ö
àô " )¨VÕ4ð
ˆð 
ð ˆùò	
s   ”+)rC   r(   r)   r  r(  r-  )r¥  rD   r¦  s   `  r4   r5  r5  ­  s<   ø€ ô �)œUŸY™Y×3Ñ3Ô4ØÐä˜YÔ'ô	ô   	Ð+MÓNÐNr6   c                 ó˜  — t        j                  | «      ry t        | t        j                  j
                  «      ry t        j                  | «      s't        j                  | «      st        d| › d| › d�«      ‚t        | d«      r| j                  «       n| } t        j                  | «      }t        j                  j                  | |¬«      S )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<   r7  rC   r(   r—   r   rŠ   rÒ   r6  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ˆ×	"Ñ	"Ô	$ÐSU€Bô
 ×
;Ñ
;¸BÓ
?€CÜ�9‰9×Ñ˜B SÐÓ)Ð)r6   c                 ó@   — t         j                  j                  | «      S )z;Wrap this torch._C.Object in a Python RecursiveScriptClass.)r(   r)   ÚRecursiveScriptClass)Ú	cpp_classs    r4   r"  r"  ä  s   € ä�9‰9×)Ñ)¨)Ó4Ð4r6   c                 ó\   — d„ }t         j                  j                  j                  | |«      S )zYWrap this torch._C.ScriptModule in a Python ScriptModule, recursively for all submodules.c                 ó"  — t         j                  j                  | j                  «      j	                  «       D ]  \  }}t        | |t        |«      «       Œ t         j                  j                  j                  | j                  j                  «       «      | _
        t        | j                  j                  «       «      D ]  \  }}|| j                  |<   Œ t        | j                  j                  «       «      D ]  \  }}|| j                  |<   Œ y r&   )r(   r,   r­   rL  rU   r!  Úwrap_cpp_moduler¾   r¿   Ú_typer  rK  rM  r   rN  r   )r;  r?   rA  rZ  r[  s        r4   rB  z wrap_cpp_module.<locals>.init_fnì  sæ   € Ü %§¡× 3Ñ 3°M×4DÑ4DÓ E× KÑ KÓ Mò 	FÑˆD�*Ü�M 4¬¸Ó)DÕEð	Fä',§x¡x×'BÑ'B×'PÑ'PØ×Ñ×"Ñ"Ó$ó(
ˆÔ$ô ! ×!1Ñ!1×!HÑ!HÓ!JÓKò 	7‰GˆC�Ø46ˆM×,Ñ,¨SÒ1ð	7ä  ×!1Ñ!1×!DÑ!DÓ!FÓGò 	3‰GˆC�Ø02ˆM×(Ñ(¨Ò-ñ	3r6   )r(   r)   r9   rI  )rA  rB  s     r4   r±  r±  é  s&   € ò
3ô �9‰9×*Ñ*×5Ñ5°jÀ'ÓJÐJr6   c                 óâ   — t        j                  |«      ry t        ||j                  «      }t        j                   j                  «       5  t        | |fd«       d d d «       |S # 1 sw Y   |S xY w)Nr£   )r<   r7  r@   r'   r(   Ú_disable_emit_hooksr  )rý   r[  Ústubs      r4   Úcompile_unbound_methodr¶  û  sf   € Ü×"Ñ" 2Ô&ØÜ�R˜Ÿ™Ó%€DÜ	×	Ñ	×	0Ñ	0Ó	2ñ Mô 	1°ÀÀÈÔL÷Mð €K÷	Mð €Kús   ÁA$Á$A.c                 ó†   ‡ ‡— ˆ ˆfd„}‰|_         ‰j                  |_        t        j                  j	                  ‰|«       |S )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                 ó˜   •— ˆfd„}t         j                  j                  j                  | |«      }t	        j
                  ‰|«      } ||Ž S )Nc                 ó  •— ‰j                   }t        |«      D ]2  }t        ||d «      }t        j                  |«      sŒ&t        | ||«       Œ4 ‰j                  «       j                  «       D ]  \  }}t        | ||«       Œ y r&   )Úpy_classrH   rB   r<   r7  r!  Úget_constantsrU   )r;  Ú
orig_classr?   rO   ræ   rý   s        €r4   rB  z7lazy_bind.<locals>.lazy_binding_method.<locals>.init_fn  s~   ø€ Ø&×/Ñ/ˆJô ˜J›ò 7�Ü˜z¨4°Ó6�Ü ×.Ñ.¨tÕ4Ü˜M¨4°Õ6ð7ð  -×:Ñ:Ó<×BÑBÓDò 4‘��eÜ˜ t¨UÕ3ñ4r6   )r(   r)   r9   rI  ÚtypesÚ
MethodType)rA  ÚargsrB  r;  rk  rý   Úunbound_methods        €€r4   Úlazy_binding_methodz&lazy_bind.<locals>.lazy_binding_method  sB   ø€ ô	4ô Ÿ	™	×7Ñ7×BÑBÀ:ÈwÓ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   )T)TF)LÚcollectionsrT  rŠ   r‘   rt   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   rb   ÚboolÚfloatÚintrp  r[   ÚdeviceÚlayoutÚdtypero   re   r,   Ú_jit_tree_viewsÚSourceRangeFactoryrx   r‹   rñ   ró   r  r  r  rÀ   r&  r-  r*  rg  rY  rÊ   rÉ   r  r‰  r(  rš  rH  rG  r5  r«  r"  r±  r¶  rÄ  r£   r6   r4   ú<module>rÖ     s~  ðã Û Û Û 
Û Û Û ã Ý +Ý %Ý -Ý <ß MÑ M÷ó õ ð *�;×)Ñ)ØÐJóÐ ð &ˆ{×%Ñ% nÐ6UÓV€òÐ ò,ò,ò(ò
ò+ð, 	Ø	ØØÙˆƒJØ	‡L�LØ	‡L�LØ	‡K�Kð	€òô&P�E—H‘H×,Ñ,×?Ñ?ô Pò
$ó2k!÷\ñ ñ< (Ó)Ð òòTóòB&ó,Iò0Yò~'ò9ò
ò@"ò
òò&ò*6"òr&òDò(Oò6*ò85ò
Kò$ó r6   