+
    ˜°³jõ5  ã                  óÚ   € ^ RI Ht ^ RIHt ^ RIHt ^ RIHt ^ RIH	t	 ^ RI
t
^ RIHtHtHtHt ]	'       d   ^ RIHt ]! R	R
7       ! R R4      4       tR R ltR R ltR R ltR R ltR# )é    )Úannotations)Údefaultdict)ÚIterable)Ú	dataclass)ÚTYPE_CHECKINGN)Úmerge_debug_infoÚmerge_operator_dictsÚSelectiveBuildOperatorÚstrip_operator_overload_name)ÚNativeFunctionT)Úfrozenc                  ó6  € ] tR t^t$ R]R&   R]R&   R]R&   R]R&   R]R	&   R
]R&   R
]R&   R]R&   ]R R l4       t]R R l4       t]R R l4       t]R R l4       t	]R R l4       t
R R ltR R ltR R ltR R ltR  R! ltR" R# ltR$ R% ltR& R' ltR(tR)# )*ÚSelectiveBuilderÚboolÚinclude_all_operatorsztuple[str, ...] | NoneÚ_debug_infoz!dict[str, SelectiveBuildOperator]Ú	operatorsúdict[str, list[str]]Úkernel_metadataÚet_kernel_metadataúset[str]Úcustom_classesÚbuild_featuresÚinclude_all_non_op_selectivesc               ó   € V ^8„  d   QhRR/# )é   Úreturnr   © )Úformats   "Úb/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torchgen/selective_build/selector.pyÚ__annotate__ÚSelectiveBuilder.__annotate__G   s   € ÷ Pñ PÐ.ñ Pó    c                 	ó0   € \         P                  R R/4      # )r   T©r   Úfrom_yaml_dictr   r#   r    Úget_nop_selectorÚ!SelectiveBuilder.get_nop_selectorF   s   € ä×.Ñ.Ð0GÈÐ/NÓOÐOr#   c               ó    € V ^8„  d   QhRRRR/# )r   Údataúdict[str, object]r   r   r   )r   s   "r    r!   r"   K   s    € ÷ Z
ñ Z
Ð.ð Z
Ð3Cñ Z
r#   c           
     	óR  € 0 Rmp\        V P                  4       4      p\        W!,
          4      ^ 8”  d0   \        RP	                  R	P                  W!,
          4      4      4      hV P                  RR
4      p\        V\        4      '       g   \        R\        V4       24      hRpRV 9   dc   V R,          p\        V\        4      '       g   \        R\        V4       24      h\        ;QJ d    . R V 4       F  NK  	  5M! R V 4       4      p/ pV P                  R/ 4      p\        V\        4      '       g   \        R\        V4       24      hVP                  4        F  w  r‰\        P                   ! W‰4      Wh&   K  	  / p
V P                  R/ 4      p\        V\        4      '       g   \        R\        V4       24      hVP                  4        F,  w  r‰V	 Uu. uF  p\#        V4      NK  	  upV
\#        V4      &   K.  	  V P                  R/ 4      p\        V\        4      '       g   \        R\        V4       24      hV P                  R. 4      p\        V\$        4      '       g   \        R\        V4       24      h\        V4      pV P                  R. 4      p\        V\$        4      '       g   \        R\        V4       24      h\        V4      pV P                  R R
4      p\        V\        4      '       g   \        R\        V4       24      h\'        VVVV
VVVV4      # u upi )r   r   Ú
debug_infor   r   r   r   r   z!Got unexpected top level keys: {}Ú,Fz1Expected 'include_all_operators' to be bool, got Nz&Expected 'debug_info' to be list, got c              3  ó8   "  € T F  p\        V4      x € K  	  R # 5i©N)Ústr)Ú.0Úxs   & r    Ú	<genexpr>Ú2SelectiveBuilder.from_yaml_dict.<locals>.<genexpr>k   s   é € Ð7©w¨!œs 1Ÿv˜v«wùs   ‚z%Expected 'operators' to be dict, got z+Expected 'kernel_metadata' to be dict, got z.Expected 'et_kernel_metadata' to be dict, got z.Expected 'custom_classes' to be Iterable, got z.Expected 'build_features' to be Iterable, got z9Expected 'include_all_non_op_selectives' to be bool, got >   r   r-   r   r   r   r   r   r   )ÚsetÚkeysÚlenÚ	Exceptionr   ÚjoinÚgetÚ
isinstancer   ÚAssertionErrorÚtypeÚlistÚtupleÚdictÚitemsr
   r&   r1   r   r   )r*   Úvalid_top_level_keysÚtop_level_keysr   r-   Údi_listr   Úoperators_dictÚkÚvr   Úkernel_metadata_dictÚdtyper   r   r   r   s   &                r    r&   ÚSelectiveBuilder.from_yaml_dictJ   s  € ò	 
Ðô ˜TŸY™Y›[Ó)ˆÜˆ~Õ4Ó5¸Ô9ÜØ3×:Ñ:Ø—H‘H˜^ÕBÓCóóð ð
 !%§¡Ð)@À%Ó HÐÜÐ/´×6Ò6Ü ØCÄDÐI^ÓD_ÐC`Ðaóð ð ˆ
Ø˜4ÔØ˜<Õ(ˆGÜ˜g¤t×,Ò,Ü$Ø<¼TÀ'»]¸OÐLóð ÷ œÑ7©wÓ7Ÿ™Ñ7©wÓ7Ó7ˆJàˆ	ØŸ™ +¨rÓ2ˆÜ˜.¬$×/Ò/Ü Ø7¼¸^Ó8LÐ7MÐNóð ð #×(Ñ(Ö*‰DˆAÜ1×@Ò@ÀÓFˆI‹Lñ +ð ˆØ#Ÿx™xÐ(9¸2Ó>ÐÜÐ.´×5Ò5Ü Ø=¼dÐCWÓ>XÐ=YÐZóð ð )×.Ñ.Ö0‰DˆAÙ?@Ó&A¹q°e¤s¨5¦z¹qÑ&AˆOœC ›FÓ#ñ 1ð "ŸX™XÐ&:¸BÓ?ÐÜÐ,¬d×3Ò3Ü Ø@ÄÐFXÓAYÐ@ZÐ[óð ð Ÿ™Ð"2°BÓ7ˆÜ˜.¬(×3Ò3Ü Ø@ÄÀnÓAUÐ@VÐWóð ô ˜^Ó,ˆàŸ™Ð"2°BÓ7ˆÜ˜.¬(×3Ò3Ü Ø@ÄÀnÓAUÐ@VÐWóð ô ˜^Ó,ˆà(,¯©Ð1PÐRWÓ(XÐ%ÜÐ7¼×>Ò>Ü ðÜÐ9Ó:Ð;ð=óð ô
  Ø!ØØØØØØØ)ó	
ð 		
ùò; 'Bs   ÇL$c               ó    € V ^8„  d   QhRRRR/# )r   Úconfig_contentsr1   r   r   r   )r   s   "r    r!   r"   ¨   s   € ÷ 9ñ 9 sð 9Ð/?ñ 9r#   c                	óX   € \         P                  ! V 4      p\        P                  V4      # r0   )ÚyamlÚ	safe_loadr   r&   )rM   Úcontentss   & r    Úfrom_yaml_strÚSelectiveBuilder.from_yaml_str§   s!   € ä—>’> /Ó2ˆÜ×.Ñ.¨xÓ8Ð8r#   c               ó    € V ^8„  d   QhRRRR/# )r   Úconfig_pathr1   r   r   r   )r   s   "r    r!   r"   ­   s   € ÷ =ñ = Cð =Ð,<ñ =r#   c                	ó¼   € \        V 4      ;_uu_ 4       p\        P                  ! V4      p\        P	                  V4      uuR R R 4       #   + '       g   i     R # ; ir0   )ÚopenrO   rP   r   r&   )rU   ÚfrQ   s   &  r    Úfrom_yaml_pathÚSelectiveBuilder.from_yaml_path¬   s:   € ä�+×Ô !Ü—~’~ aÓ(ˆHÜ#×2Ñ2°8Ó<÷ ××Óús   ”+A
Á
A	c               ó(   € V ^8„  d   QhRRRRRRRR/# )r   Ú
allow_listr   Úis_root_operatorr   Úis_used_for_trainingr   r   r   )r   s   "r    r!   r"   ³   s,   € ÷ 
ñ 
Øð
Ø04ð
ØLPð
à	ñ
r#   c           	     	ób   € / pV  F  pR VRVRVRR/W4&   K  	  \         P                  RVRR/4      # )Únamer]   r^   Úinclude_all_overloadsTr   r   r%   )r\   r]   r^   r   Úops   &&&  r    Ú&from_legacy_op_registration_allow_listÚ7SelectiveBuilder.from_legacy_op_registration_allow_list²   sV   € ð ˆ	ÛˆBà˜Ø"Ð$4Ø&Ð(<Ø'¨ð	ˆI‹Mñ ô  ×.Ñ.à˜YØ/°ðó
ð 	
r#   c               ó    € V ^8„  d   QhRRRR/# ©r   r`   r1   r   r   r   )r   s   "r    r!   r"   Å   s   € ÷ Uñ U¨ð U°ñ Ur#   c                	óÊ   € V P                   '       d   R # WP                  9   d   R # \        V4      pWP                  9   ;'       d    V P                  V,          P                  # ©T)r   r   r   ra   )Úselfr`   s   &&r    Úis_operator_selectedÚ%SelectiveBuilder.is_operator_selectedÅ   sL   € Ø×%×%Ð%Ùà—>‘>Ô!ÙÜ+¨DÓ1ˆØ—~‘~Ñ%×TÐT¨$¯.©.¸Õ*>×*TÑ*TÐTr#   c               ó    € V ^8„  d   QhRRRR/# ©r   Úfuncr   r   r   r   )r   s   "r    r!   r"   Î   s   € ÷ 2ñ 2°ð 2À4ñ 2r#   c                	ó:   € \        V4      pV P                  V4      # r0   )Úop_name_from_native_functionrj   ©ri   rn   Úop_names   && r    Úis_native_function_selectedÚ,SelectiveBuilder.is_native_function_selectedÎ   s   € Ü.¨tÓ4ˆØ×(Ñ(¨Ó1Ð1r#   c               ó    € V ^8„  d   QhRRRR/# rf   r   )r   s   "r    r!   r"   Ò   s   € ÷ 
ñ 
°cð 
¸dñ 
r#   c                	óŽ  € V P                  V4      '       g   R # V P                  '       d   R# \        RR R R RR7      pTpWP                  9   d   V P                  V,          p\	        V4      pTpWP                  9   d   V P                  V,          pVP
                  ;'       g!    VP                  ;'       d    VP
                  # )FTÚ N)r`   r]   r^   ra   r   )rj   r   r
   r   r   r^   ra   )ri   r`   Únot_training_oprb   Úbase_ops   &&   r    Ú!is_operator_selected_for_trainingÚ2SelectiveBuilder.is_operator_selected_for_trainingÒ   s³   € Ø×(Ñ(¨×.Ò.ÙØ×%×%Ð%Ùä0ØØ"Ø!&Ø"'Øô
ˆð ˆØ—>‘>Ô!Ø—‘ Õ%ˆBä+¨DÓ1ˆØ!ˆØ—>‘>Ô!Ø—n‘n TÕ*ˆGà×&Ñ&÷ 
ð 
Ø×)Ñ)×JÐJ¨g×.JÑ.Jð	
r#   c               ó    € V ^8„  d   QhRRRR/# rm   r   )r   s   "r    r!   r"   ì   s   € ÷ ?ñ ?¸^ð ?ÐPTñ ?r#   c                	ó:   € \        V4      pV P                  V4      # r0   )rp   rz   rq   s   && r    Ú(is_native_function_selected_for_trainingÚ9SelectiveBuilder.is_native_function_selected_for_trainingì   s   € Ü.¨tÓ4ˆØ×5Ñ5°gÓ>Ð>r#   c               ó    € V ^8„  d   QhRRRR/# rf   r   )r   s   "r    r!   r"   ð   s   € ÷ Jñ J Sð J¨Tñ Jr#   c                	óZ  € V P                  V4      '       g   R # V P                  '       d   R# WP                  9   d    V P                  V,          pVP                  # \	        V4      pWP                  9  d   R # V P                  V,          pVP
                  ;'       d    VP                  # )FT)rj   r   r   r]   r   ra   )ri   r`   rb   ry   s   &&  r    r]   Ú!SelectiveBuilder.is_root_operatorð   sˆ   € Ø×(Ñ(¨×.Ò.ÙØ×%×%Ð%Ùà—>‘>Ô!Ø)-¯©¸Õ)=ˆBØ×&Ñ&Ð&Ü+¨DÓ1ˆØ—~‘~Ô%ÙØ*.¯.©.¸Õ*>ˆØ×,Ñ,×IÐI°×1IÑ1IÐIr#   c               ó$   € V ^8„  d   QhRRRRRR/# )r   Ú
kernel_tagr1   rJ   r   r   r   )r   s   "r    r!   r"   ÿ   s!   € ÷ 
ñ 
°3ð 
¸sð 
Àtñ 
r#   c                	ó¤   € V P                   '       g   V P                  '       d   R # WP                  9   ;'       d    W P                  V,          9   # rh   )r   r   r   )ri   r„   rJ   s   &&&r    Úis_kernel_dtype_selectedÚ)SelectiveBuilder.is_kernel_dtype_selectedÿ   sJ   € Ø×%×%Ð%¨×)K×)KÐ)KÙð ×.Ñ.Ñ.÷ :ð :Ø×-Ñ-¨jÕ9Ñ9ð	
r#   c               ó$   € V ^8„  d   QhRRRRRR/# )r   rr   r1   Ú
kernel_keyz	list[str]r   r   )r   s   "r    r!   r"     s!   € ÷  ñ  ¨sð  À	ð  Èiñ  r#   c                ó¾  € WP                   9  d   V P                  '       d   V# . # \        4       pV P                   V,           FŒ  pRpV FT  pVR8w  g   K  VP                  R4      ^,          VP                  R4      ^,          8X  g   KA  VP	                  V4       Rp M	  V'       d   Ki  RV9  d   \        R4      hVP	                  R4       KŽ  	  \        V4      # )z6
Return a list of kernel keys that cover the used ops
FÚdefaultÚ/TzMissing kernel for the model)r   r   r6   ÚsplitÚaddr9   r?   )ri   rr   r‰   Ú
result_setÚmodel_kernel_keysÚ	key_foundÚkeys   &&&    r    Úet_get_selected_kernelsÚ(SelectiveBuilder.et_get_selected_kernels  sÐ   € ð
 ×1Ñ1Ô1Ø!%×!;×!;Ð!;�:ÐCÀÐCô “Uˆ
à!%×!8Ñ!8¸×!AÐ!AÐØˆIÛ!�ð ˜9Ö$ØŸ	™	 #› qÕ)Ð->×-DÑ-DÀSÓ-IÈ!Õ-LÖLà—N‘N 3Ô'Ø $�IÙñ "÷ ‘9Ø JÔ.Ü#Ð$BÓCÐCà—N‘N 9Ö-ñ "Bô" �JÓÐr#   c               ó   € V ^8„  d   QhRR/# )r   r   r+   r   )r   s   "r    r!   r"   &  s   € ÷ ñ Ð*ñ r#   c                	óü  € R V P                   RV P                  /p/ pV P                  P                  4        F  w  r4VP	                  4       W#&   K  	  W!R&   V P
                  e   \        V P
                  4      VR&   V P                  P                  4        UUu/ uF  w  rVV\        V4      bK  	  uppVR&   V P                  VR&   \        V P                  4      VR&   \        V P                  4      VR&   V# u uppi )r   r   r   r-   r   r   r   r   )r   r   r   rB   Úto_dictr   Úsortedr   r   r   r   )ri   Úretr   rr   rb   rG   rH   s   &      r    r—   ÚSelectiveBuilder.to_dict&  sú   € à+¨T×-OÑ-OØ# T×%?Ñ%?ð"
ˆð ˆ	ØŸ>™>×/Ñ/Ö1‰KˆGØ!#§¡£ˆIÓñ 2à$ˆKÑà×ÑÒ'Ü & t×'7Ñ'7Ó 8ˆC�Ñð (,×';Ñ';×'AÑ'AÔ'Cô"
Ù'C™V˜aˆAŒv�a‹yŠLÑ'Cò"
ˆÐÑð %)×$;Ñ$;ˆÐ Ñ!ä & t×':Ñ':Ó ;ˆÐÑä & t×':Ñ':Ó ;ˆÐÑàˆ
ùó"
s   ÂC8r   N)Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__annotations__Ústaticmethodr'   r&   rR   rY   rc   rj   rs   rz   r~   r]   r†   r“   r—   Ú__static_attributes__r   r#   r    r   r      sØ   ‡ ð  Óð (Ó'ð 1Ó0ð *Ó)ð
 -Ó,ð
 Óð
 Óð $(Ó'àôPó ðPð ôZ
ó ðZ
ðx ô9ó ð9ð ô=ó ð=ð
 ô
ó ð
õ$Uõ2õ
õ4?õJõ
õ ÷<ñ r#   r   c               ó$   € V ^8„  d   QhRRRRRR/# ©r   Úlhsr   Úrhsr   r   )r   s   "r    r!   r!   @  s&   € ÷ ñ Ø	ðà	ðð ñr#   c                óö   € / p\        V P                  4       4      \        VP                  4       4      ,            F;  w  r4\        V4      pW29   d   V\        W#,          4      ,          p\        V4      W#&   K=  	  V# r0   )r?   rB   r6   )r¤   r¥   r   Útag_nameÚdtypesÚdtypes_copys   &&    r    Úmerge_kernel_metadatarª   @  sk   € ð -/€OÜ  §¡£Ó-´°S·Y±Y³[Ó0A×AÐAÑˆÜ˜&“kˆØÔ&Øœ3˜Õ8Ó9Õ9ˆKä$(¨Ó$5ˆÓ!ñ Bð Ðr#   c               ó$   € V ^8„  d   QhRRRRRR/# r£   r   )r   s   "r    r!   r!   O  s+   € ÷ 	Mñ 	MØ	ð	Mà	ð	Mð ñ	Mr#   c                ó   € \        \        4      p\        V P                  4       4      \        VP                  4       4      ,            FQ  pW#,          P	                  V P                  V. 4      4       W#,          P	                  VP                  V. 4      4       KS  	  VP                  4        UUu/ uF  w  r4V\        V4      bK  	  upp# u uppi r0   )r   r6   r?   r7   Úupdater;   rB   r˜   )r¤   r¥   Úmerge_et_kernel_metadatarb   Úvals   &&   r    r®   r®   O  sž   € ô 5@ÄÓ4DÐÜ�3—8‘8“:Ó¤ c§h¡h£jÓ!1×1Ð1ˆØ Õ$×+Ñ+¨C¯G©G°B¸«OÔ<Ø Õ$×+Ñ+¨C¯G©G°B¸«OÖ<ñ 2ð ,D×+IÑ+IÔ+KÔLÑ+K¡ ˆB”�s“ŠOÑ+KÒLÐLùÓLs   Â/C
c               ó$   € V ^8„  d   QhRRRRRR/# )r   r¤   r   r¥   r   r   )r   s   "r    r!   r!   [  s$   € ÷ ñ Ø	ðØ 0ðàñr#   c           
     ó:  € V P                   ;'       g    VP                   p\        V P                  VP                  4      p\        V P                  VP                  4      p\        V P                  VP                  4      p\        V P                  VP                  4      pV P                  ;'       g    VP                  pV P                  P                  VP                  4      pV P                  P                  VP                  4      p	\        VVVVVVV	V4      # r0   )r   r   r   r	   r   rª   r   r®   r   r   r   Úunionr   r   )
r¤   r¥   r   r-   r   r   r   r   r   r   s
   &&        r    Úcombine_selective_buildersr³   [  s÷   € ð  ×5Ñ5×RÐR¸×9RÑ9RÐÜ! #§/¡/°3·?±?ÓC€JÜ$ S§]¡]°C·M±MÓB€IÜ+¨C×,?Ñ,?À×ATÑATÓU€OÜ1Ø×Ñ × 6Ñ 6óÐð 	×)Ñ)×NÐN¨S×-NÑ-Nð "ð ×'Ñ'×-Ñ-¨c×.@Ñ.@ÓA€NØ×'Ñ'×-Ñ-¨c×.@Ñ.@ÓA€NÜØØØØØØØØ%ó	ð 	r#   c               ó    € V ^8„  d   QhRRRR/# )r   rX   r   r   r1   r   )r   s   "r    r!   r!   v  s   € ÷ +ñ + Nð +°sñ +r#   c                óL   € V P                    R V P                  P                   2# )z::)Ú	namespacern   r`   )rX   s   &r    rp   rp   v  s!   € ð �k‰kˆ]˜"˜QŸV™VŸ[™[˜MÐ*Ð*r#   )Ú
__future__r   Úcollectionsr   Úcollections.abcr   Údataclassesr   Útypingr   rO   Ú!torchgen.selective_build.operatorr   r	   r
   r   Útorchgen.modelr   r   rª   r®   r³   rp   r   r#   r    Ú<module>r¾      s`   ðÝ "å #Ý $Ý !Ý  ã ÷ó ÷ Ý-ñ �$Ô÷_ð _ó ð_õD	õ	Mõ÷6+r#   