+
    –°³j­  ã                   óÀ   € ^ RI Ht ^ RIHtHtHt ^ RIHt ^ RIH	t	 ^ RI
Ht ^RIHt ^RIHt ^RIHt ^R	IHt ]'       d   ^ R
IHt ^ RIHt . ROt ! R R]4      tR# )é    )Úgetattr_static)ÚAnyÚTYPE_CHECKINGÚ	TypeGuard)ÚSource©Ú
SDPAParams©ÚProxy)Úcreate_call_function)Úunimplemented)Ú
AttrSource)ÚVariableTracker)Ú	PyCodegen)ÚInstructionTranslatorBasec                   óÂ   a a€ ] tR t^t oRt]V3R lR l4       tV3R lV 3R lltV3R lR ltV3R lR	 lt	V3R
 lR lt
V3R lR lt]V3R lR l4       tRtVtV ;t# )ÚSDPAParamsVariablezaRepresents the c++ params struct for scaled dot product attention.
This is a read-only container.c                ó0   <€ V ^8„  d   QhRRRS[ RS[RS[/# )é   Útxr   ÚvalueÚsourceÚreturn)r   r   r   )ÚformatÚ__classdict__s   "€Ú]/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_dynamo/variables/sdpa.pyÚ__annotate__ÚSDPAParamsVariable.__annotate__"   s3   ø€ ÷ 	Vñ 	VØ'ð	VÙ03ð	VÙ=Cð	Vá	ñ	Vó    c                óÊ   € ^RI Hp \         Uu. uF-  p\        P                  ! V \        W4      \        W$4      4      NK/  	  ppV! \        4      P                  W/ 4      # u upi )é   )ÚTorchInGraphFunctionVariable)	Útorchr"   ÚPARAM_NAMESr   ÚbuildÚgetattrr   r	   Úcall_function)r   r   r   r"   ÚpÚparamss   &&&   r   ÚcreateÚSDPAParamsVariable.create!   sb   € õ 	8õ !ó
á �ô ×!Ò! "¤g¨eÓ&7¼ÀFÓ9NÖOÙ ð 	ð 
ñ ,¬JÓ7×EÑEÀbÐRTÓUÐUùò	
s   �3A c                ó@   <€ V ^8„  d   QhRS[ RS[S[,          RS[RR/# )r   ÚproxyÚ
param_varsÚkwargsr   N)r   Úlistr   r   )r   r   s   "€r   r   r   -   s2   ø€ ÷ #ñ #Ùð#Ù(,©_Õ(=ð#ÙILð#à	ñ#r   c                ó@   <€ Wn         W n        \        SV `  ! R/ VB  R # )N© )r-   r.   ÚsuperÚ__init__)Úselfr-   r.   r/   Ú	__class__s   &&&,€r   r4   ÚSDPAParamsVariable.__init__-   s   ø€ ð Œ
Ø$ŒÜ‰ÒÑ"˜6Ô"r   c                ó    <€ V ^8„  d   QhRS[ /# ©r   r   )Útype)r   r   s   "€r   r   r   4   s   ø€ ÷ ñ ™Tñ r   c                ó   € \         # ©Nr   ©r5   s   &r   Úpython_typeÚSDPAParamsVariable.python_type4   s   € ÜÐr   c                ó"   <€ V ^8„  d   QhRRRR/# )r   Úcodegenr   r   Nr2   )r   r   s   "€r   r   r   7   s   ø€ ÷ Qñ Q ;ð Q°4ñ Qr   c                ó(  a€ V P                   e   \        R4      hV P                  f   \        R4      hSP                  V3R l4       SP	                  V P                  4       SP                  \        \        V P                  4      R4      4       R # )Nz>SDPAParamsVariable should not have a source during reconstructz.SDPAParamsVariable.param_vars must not be Nonec                  ó(   <€ S P                  R R4      # )ztorch._CÚ_SDPAParams)Úload_import_from)rA   s   €r   Ú<lambda>Ú0SDPAParamsVariable.reconstruct.<locals>.<lambda>?   s   ø€ �G×,Ñ,¨Z¸ÔGr   F)r   ÚAssertionErrorr.   Úadd_push_nullÚforeachÚextend_outputr   Úlen)r5   rA   s   &fr   ÚreconstructÚSDPAParamsVariable.reconstruct7   sw   ø€ Ø�;‰;Ò"Ü ØPóð ð �?‰?Ò"Ü Ð!QÓRÐRØ×ÑÜGô	
ð 	�‰˜Ÿ™Ô(Ø×ÑÔ2´3°t·±Ó3GÈÓOÖPr   c                ó    <€ V ^8„  d   QhRS[ /# r9   r
   )r   r   s   "€r   r   r   D   s   ø€ ÷ ñ ™%ñ r   c                ó   € V P                   # r<   )r-   r=   s   &r   Úas_proxyÚSDPAParamsVariable.as_proxyD   s   € Ø�z‰zÐr   c                ó*   <€ V ^8„  d   QhRRRS[ RS[/# )r   r   r   Únamer   )Ústrr   )r   r   s   "€r   r   r   G   s%   ø€ ÷ 5ñ 5Ø-ð5Ù58ð5á	ñ5r   c                ó†  € ^ RI p^RIHp ^RIHp  \        VP                  P                  V4       VP                  V P                  4       V4      pV P                   e   V! W\#        V P                   V4      R	7      # V! WR
7      #   \         d/    ^ RI	H
u Hp \        RRT 2RT R2. TP                  OR7        L�i ; i)r   N)Úwrap_fx_proxy)ÚGetAttrVariablez*unsupported torch._C._SDPAParams attributezname: zUnable to fetch attribute z from torch._C._SDPAParams.)Úgb_typeÚcontextÚexplanationÚhints)r   r-   r   )r   r-   )Útorch._CÚbuilderrW   ÚmiscrX   r   Ú_CrD   ÚAttributeErrorÚtorch._dynamo.graph_break_hintsÚ_dynamoÚgraph_break_hintsr   Ú
USER_ERRORÚcreate_getattr_proxyrQ   r   r   )r5   r   rT   r#   rW   rX   rd   r-   s   &&&     r   Útp_getattro_implÚ#SDPAParamsVariable.tp_getattro_implG   s¿   € ó 	å*Ý)ð	Ü˜5Ÿ8™8×/Ñ/°Ô6ð  ×4Ñ4°T·]±]³_ÀdÓKˆØ�;‰;Ò"Ù Ø¬:°d·k±kÀ4Ó+Hôð ñ ! BÔ4Ð4øô% ô 
	ßGÐGäØDØ   ˜Ø8¸¸Ð>YÐZðØ&×1Ñ1ð÷	ð
	ús   ’ B Â6C Â?C c                ó4   <€ V ^8„  d   QhRS[ RS[R,          /# )r   r   r   r	   )r   r   )r   r   s   "€r   r   r   f   s    ø€ ÷ #ñ #™cð #¡i°Õ&=ñ #r   c                ó   € V \         J # r<   r   )r   s   &r   Úis_sdpa_paramsÚ!SDPAParamsVariable.is_sdpa_paramse   s   € àœ
Ð"Ð"r   )r.   r-   )Ú__name__Ú
__module__Ú__qualname__Ú__firstlineno__Ú__doc__Ústaticmethodr*   r4   r>   rM   rQ   rg   rk   Ú__static_attributes__Ú__classdictcell__Ú__classcell__)r6   r   s   @@r   r   r      sm   ù‡ € ñ&ð ÷	Vó ð	V÷#ó #÷ð ÷Qð Q÷ð ÷5ð 5ð< ÷#ó ÷#ð #r   r   N)ÚqueryÚkeyr   Ú	attn_maskÚdropoutÚ	is_causalÚ
enable_gqa)Úinspectr   Útypingr   r   r   Útorch._guardsr   Útorch.backends.cudar	   Útorch.fx.proxyr   Úbytecode_transformationr   Úexcr   r   r   Úbaser   Útorch._dynamo.codegenr   Útorch._dynamo.symbolic_convertr   r$   r   r2   r   r   Ú<module>r†      sC   ðÝ "ß 0Ñ 0å  Ý *Ý  å :Ý Ý Ý !÷ Ý/ÝHò€ôJ#˜ö J#r   