+
    ˜°³jt$  ã                   óÔ   € ^ RI t ^ RIt^ RIt^ RIt^ RIt^ RIHt ^ RIt^ RI	H
t ^ RIt^ RIHu Ht ^ RIHt R t]! RR7      R 4       tRR	/R
 R llt ! R R4      t ! R R4      tR# )é    N)Údefaultdict)ÚStorageWeakRefc                 ó   a € V 3R lpV# )zcLazily wrap a function with torch.compile on the first call

This avoids eagerly importing dynamo.
c                 óJ   <a € \         P                  ! S 4      VV 3R  l4       pV# )c                  ó¤   <€ \         P                  ! S3/ SB p\        P                  ! S4      ! V4      \	        4       SP
                  &   V! V / VB # ©N)ÚtorchÚcompileÚ	functoolsÚwrapsÚglobalsÚ__name__)ÚargsÚkwargsÚcompiled_fnÚcompile_kwargsÚfns   *, €€Ú[/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/utils/_content_store.pyÚcompile_hookÚ7lazy_compile.<locals>.decorate_fn.<locals>.compile_hook4   sD   ø€ äŸ-š-¨Ñ=¨nÑ=ˆKÜ%.§_¢_°RÔ%8¸Ó%EŒG‹I�b—k‘kÑ"Ù Ð/¨Ñ/Ð/ó    )r   r   )r   r   r   s   f €r   Údecorate_fnÚ!lazy_compile.<locals>.decorate_fn3   s&   ù€ Ü	�Š˜Ó	õ	0ó 
ð	0ð
 Ðr   © )r   r   s   l r   Úlazy_compiler   -   s   ø€ õð Ðr   T)Údynamicc                 óì  € \         P                  ! RRV P                  V P                  \         P                  R7      P                  4       pVR,          ^,           P                  4       p\         P                  ! RRV P                  V P                  \         P                  R7      P                  4       P                  4       p\        P                  ! W,          V,           P                  4       ^ .4      # )é   )ÚdeviceÚdtypel        i   €iÿÿÿ)
r	   ÚrandintÚshaper   Úint32ÚabsÚlongÚprimsÚxor_sumÚint)ÚxÚaÚbs   &  r   Úhash_storage_kernelr,   C   s¡   € ô
 	�ŠØ�%˜Ÿ™¨¯©¼¿¹ô	ç	�cƒeð ð ˆy�/˜QÕ	×$Ñ$Ó&€Aä�Š�h  q§w¡w°q·x±xÄuÇ{Á{ÔSß	‰‹ß	‰‹ð ô �=Š=˜!�% !�)Ÿ™Ó*¨Q¨CÓ0Ð0r   Ústable_hashFc                óP   € V ^8„  d   QhR\         P                  R\        R\        /# )r   Ústorager-   Úreturn)r	   ÚUntypedStorageÚboolÚstr)Úformats   "r   Ú__annotate__r5   ^   s(   € ÷ ,#ñ ,#œ%×.Ñ.ð ,#Äð ,#ÔQTñ ,#r   c                ó†  € ^ RI p^ RIHp V P                  P                  pV'       g   V! V4      '       g‹   V P                  4       p\        P                  VP                  4       ,          P                  VP                  4       4      p\        P                  ! RR7      pVP                  V4       VP                  4       # VR8X  d   VP                  P                   pM¢VR8X  d3   VP"                  P$                  V P                  P&                  ,          pMiVR8X  d   VP(                  P+                  4       pMGVR8X  d3   VP,                  P$                  V P                  P&                  ,          pM\/        R	V 24      hVP1                  4       p	 VP3                  ^ 4       VP4                  ! ^ VP6                  V P                  R
7      P9                  V 4      p
V
P;                  4       ) ^,          pV^ 8”  d   \<        P>                  ! V
^ V3R^ 4      p
V
PA                  VPB                  4      p
^p\E        V4       Uu. uF  p\G        V
4      PI                  4       NK  	  pp\J        PL                  ! RRV,          ,           .VO5!  PO                  4       VPQ                  V	4       # u upi   TPQ                  T	4       i ; i)r   N)Úis_compile_supportedF)ÚusedforsecurityÚcpuÚcudaÚmpsÚxpuzunhandled device type ©r    r   ÚconstantÚ>Úi))Útorch._dynamoÚtorch._dynamo.utilsr7   r   Útyper9   ÚctypesÚc_byteÚnbytesÚfrom_addressÚdata_ptrÚhashlibÚsha1ÚupdateÚ	hexdigestÚ_CÚdefault_generatorr:   Údefault_generatorsÚindexr;   Ú_get_default_mps_generatorr<   ÚAssertionErrorÚ	get_stateÚmanual_seedÚemptyÚuint8Úset_ÚnumelÚFÚpadÚviewr#   Úranger,   ÚitemÚstructÚpackÚhexÚ	set_state)r/   r-   r	   r7   Údevice_typeÚcpu_storageÚbufrJ   Ú	generatorÚstater)   rZ   ÚITERÚ_Úcss   &$             r   Úhash_storagerj   ^   s  € ÛÝ8à—.‘.×%Ñ%€KßÑ.¨{×;Ò;Ø—k‘k“mˆô �}‰}˜{×1Ñ1Ó3Õ3×AÑAØ× Ñ Ó"ó
ˆô �|Š|¨EÔ2ˆØ�‰�CÔØ�~‰~ÓÐð �eÔØ—H‘H×.Ñ.‰	Ø	˜Ô	Ø—J‘J×1Ñ1°'·.±.×2FÑ2FÕG‰	Ø	˜Ô	Ø—I‘I×8Ñ8Ó:‰	Ø	˜Ô	Ø—I‘I×0Ñ0°·±×1EÑ1EÕF‰	äÐ5°k°]ÐCÓDÐDØ×ÑÓ!€Eð#Ø×Ñ˜aÔ Ø�KŠK˜ §¡°W·^±^ÔD×IÑIÈ'ÓRˆð �w‰w‹yˆj˜1�nˆØ�Œ7Ü—’�a˜!˜S˜ :¨qÓ1ˆAØ�F‰F�5—;‘;Óˆð ˆÜ5:¸4´[ÓA±[°Ô! !Ó$×)Ñ)Ö+±[ˆÐAÜ�{Š{˜3  t¥Õ+Ð1¨bÓ1×5Ñ5Ó7à×Ñ˜EÕ"ùò Bøð 	×Ñ˜EÕ"ús   ÆB-J- È?"J(É!5J- Ê(J- Ê-K c                   óp   a € ] tR t^�t o RV 3R lR lltV 3R lR ltRV 3R lR lltV 3R lR	 ltR
tV t	R# )ÚContentStoreWriterc                ó*   <€ V ^8„  d   QhRS[ RS[RR/# )r   Úlocr-   r0   N)r3   r2   )r4   Ú__classdict__s   "€r   r5   ÚContentStoreWriter.__annotate__”   s"   ø€ ÷ 'ñ '™Cð '©dð '¸tñ 'r   c                ó<   € Wn         \        4       V n        W n        R # r   )rn   ÚsetÚseen_storage_hashesr-   )Úselfrn   r-   s   &&&r   Ú__init__ÚContentStoreWriter.__init__”   s   € ØŒÜ-0«UˆÔ Ø&Ör   c                ó:   <€ V ^8„  d   QhRS[ P                  RS[/# )r   r/   r0   )r	   r1   r3   )r4   ro   s   "€r   r5   rp   š   s!   ø€ ÷ ñ ¡U×%9Ñ%9ð ¹cñ r   c                óÈ  € \        WP                  R 7      pW P                  9   d   V# \        P                  P                  V P                  R4      p\        P                  ! VRR7       \        P                  P                  W24      p\        P                  P                  V4      '       d   V# \        P                  ! W4       V P                  P                  V4       V# )©r-   ÚstoragesT©Úexist_ok)rj   r-   rs   ÚosÚpathÚjoinrn   ÚmakedirsÚexistsr	   ÚsaveÚadd)rt   r/   ÚhÚ	subfolderÚtargets   &&   r   Úwrite_storageÚ ContentStoreWriter.write_storageš   sš   € Ü˜×.>Ñ.>Ô?ˆØ×(Ñ(Ô(ØˆHô —G‘G—L‘L §¡¨:Ó6ˆ	Ü
�Š�I¨Õ-Ü—‘—‘˜iÓ+ˆÜ�7‰7�>‰>˜&×!Ò!ØˆHÜ�
Š
�7Ô#Ø× Ñ ×$Ñ$ QÔ'Øˆr   Nc                ó4   <€ V ^8„  d   QhRS[ P                  /# )r   Út)r	   ÚTensor)r4   ro   s   "€r   r5   rp   ©   s   ø€ ÷ 

ñ 

©¯©ñ 

r   c                ó  € Vf&   \        VP                  4       V P                  R7      pVP                  VVP	                  4       \        VP                  4      VP                  4       \        P                  P                  V4      3# )Nry   )rj   Úuntyped_storager-   r    Ústorage_offsetÚtupler"   Ústrider	   Ú_utilsÚget_tensor_metadata)rt   rŠ   r„   s   &&&r   Úcompute_tensor_metadataÚ*ContentStoreWriter.compute_tensor_metadata©   sh   € ØŠ9Ü˜Q×.Ñ.Ó0¸d×>NÑ>NÔOˆAà�G‰GØØ×ÑÓÜ�!—'‘'‹NØ�H‰H‹JÜ�L‰L×,Ñ,¨QÓ/ð
ð 	
r   c                ó>   <€ V ^8„  d   QhRS[ RS[P                  RR/# )r   ÚnamerŠ   r0   N©r3   r	   r‹   )r4   ro   s   "€r   r5   rp   µ   s&   ø€ ÷ 8ñ 8¡ð 8©¯©ð 8¸$ñ 8r   c                óœ  € VP                  4       pV P                  V4      p\        P                  P	                  V4      w  rVV P                  W$R 7      p\        P                  P                  V P                  RV4      p\        P                  ! VRR7       \        P                  ! V\        P                  P                  W†4      4       R# ))r„   ÚtensorsTr{   N)r�   r‡   r}   r~   Úsplitr“   r   rn   r€   r	   r‚   )	rt   r–   rŠ   r/   r„   ÚdÚfÚpayloadr…   s	   &&&      r   Úwrite_tensorÚContentStoreWriter.write_tensorµ   sŠ   € Ø×#Ñ#Ó%ˆØ×Ñ˜wÓ'ˆä�w‰w�}‰}˜TÓ"‰ˆØ×.Ñ.¨qÐ.Ó6ˆÜ—G‘G—L‘L §¡¨9°aÓ8ˆ	Ü
�Š�I¨Õ-Ü�
Š
�7œBŸG™GŸL™L¨Ó6Ö7r   )rn   rs   r-   )Fr   )
r   Ú
__module__Ú__qualname__Ú__firstlineno__ru   r‡   r“   rž   Ú__static_attributes__Ú__classdictcell__©ro   s   @r   rl   rl   �   s-   ø‡ € ÷'ò '÷ð ÷

ò 

÷8ö 8r   rl   c                   ó€   a € ] tR t^Àt o RR/V 3R lR lltRR/V 3R lR lltV 3R	 lR
 ltRR/V 3R lR lltRtV t	R# )ÚContentStoreReaderÚcacheTc                ó$   <€ V ^8„  d   QhRS[ RR/# )r   rn   r0   N©r3   )r4   ro   s   "€r   r5   ÚContentStoreReader.__annotate__Á   s   ø€ ÷ 3ñ 3™Cð 3°4ñ 3r   c               ó\   € Wn         R V n        V'       d   \        \        4      V n        R # R # r   )rn   Ústorage_cacher   Údict)rt   rn   r¨   s   &&$r   ru   ÚContentStoreReader.__init__Á   s+   € ØŒð ð 	Ô÷ Ü!,¬TÓ!2ˆDÖñ r   r   Nc                ó:   <€ V ^8„  d   QhRS[ RS[P                  /# )r   r„   r0   )r3   r	   r1   )r4   ro   s   "€r   r5   r«   É   s!   ø€ ÷ ñ ™cð ±e×6JÑ6Jñ r   c          	     ór  € Ve   \         P                  ! V4      pV P                  e#   V P                  V,          P                  V4      MR pVe0   \         P                  P                  VP                  4      pVe   V# \         P                  ! \        P                  P                  V P                  RV4      RVR7      P                  pVf;   \        RV R\        P                  P                  V P                  R4       R24      hV P                  e    \        V4      V P                  V,          V&   V# )Nrz   T)Úweights_onlyÚmap_locationzexpected storage for hash z in z
, got None)r	   r   r­   Úgetr1   Ú_new_with_weak_ptrÚcdataÚloadr}   r~   r   rn   Ú_untyped_storagerR   r   )rt   r„   r   ÚwsÚss   &&$  r   Úread_storageÚContentStoreReader.read_storageÉ   s  € ØÒÜ—\’\ &Ó)ˆFð ×!Ñ!Ò-ð ×Ñ˜vÕ&×*Ñ*¨1Ô-àð 	ð Š>Ü×$Ñ$×7Ñ7¸¿¹ÓAˆAØŠ}Ø�Ü�JŠJÜ�G‰G�L‰L˜Ÿ™ :¨qÓ1ØØô
÷ Ñ
ð	 	
ð
 Š9Ü Ø,¨Q¨C¨t´B·G±G·L±LÀÇÁÈ:Ó4VÐ3WÐWaÐbóð ð ×ÑÒ)Ü,:¸1Ó,=ˆD×Ñ˜vÕ& qÑ)Øˆr   c                ó    <€ V ^8„  d   QhRS[ /# )r   r–   rª   )r4   ro   s   "€r   r5   r«   ã   s   ø€ ÷ 1ñ 1©ñ 1r   c                óè   € \         P                  P                  V P                  R V4      p\         P                  P	                  V4      '       g   \        V4      h\        P                  ! VRR7      # )r™   T)r²   )r}   r~   r   rn   r�   ÚFileNotFoundErrorr	   r·   )rt   r–   r   s   && r   Úread_tensor_metadataÚ'ContentStoreReader.read_tensor_metadataã   sK   € Ü�W‰W�\‰\˜$Ÿ(™( I¨tÓ4ˆÜ�w‰w�~‰~˜b×!Ò!Ü# BÓ'Ð'Ü�zŠz˜"¨4Ô0Ð0r   c                ó:   <€ V ^8„  d   QhRS[ RS[P                  /# )r   r–   r0   r—   )r4   ro   s   "€r   r5   r«   é   s   ø€ ÷ ñ ¡ð ¹¿¹ñ r   c               óú   € V P                  V4      w  r4rVrxV P                  WBR 7      p	\        P                  ! . W9P                  R7      p
V
P                  W•Wg4       \        P                  P                  W¨4       V
# ))r   r=   )rÀ   r»   r	   Útensorr   rW   r‘   Úset_tensor_metadata)rt   r–   r   r    r„   rŽ   Úsizer�   Úmetadatar/   rŠ   s   &&$        r   Úread_tensorÚContentStoreReader.read_tensoré   sj   € Ø;?×;TÑ;TØó<
Ñ8ˆ�.¨ð ×#Ñ# AÐ#Ó5ˆÜ�LŠL˜ 5·±Ô@ˆØ	�‰ˆw¨Ô5Ü�‰×(Ñ(¨Ô5Øˆr   )rn   r­   )
r   r    r¡   r¢   ru   r»   rÀ   rÈ   r£   r¤   r¥   s   @r   r§   r§   À   sG   ø‡ € ð3¨$÷ 3ò 3ð¨T÷ ò ÷41ð 1ð¨t÷ ÷ ð r   r§   )rD   r   rI   Úos.pathr}   r^   Úcollectionsr   r	   Útorch._primsÚ_primsr&   Útorch._utilsÚtorch.nn.functionalÚnnÚ
functionalrY   Ú torch.multiprocessing.reductionsr   r   r,   rj   rl   r§   r   r   r   Ú<module>rÓ      sk   ðó> Û Û Û Û Ý #ã Ý Û ß Ð Ý ;òñ, �dÔñ1ó ð1ð4,#Àu÷ ,#÷^08ñ 08÷f1ó 1r   