+
    ˜°³j@Y  ã            "       óä
  € ^ RI t ^ RIt^ RIHt ^ RIHt ^ RIHt ^ RI	H
t
HtHt ^ RIHt ^ RIHt ^ RIHtHtHt ^ R	IHtHt ^ R
IHt ^ RIHtHt ^ RIHt R tR tR t R t!R t"R t#. RNRNRNRNRNRNRNRNRNRNRNRNRNR NR!NR"NR#NR$NR%NR&NR'NR(NR)NR*NR+NR,NR-NR.NR/NR0NR1NR2NR3NR4NR5Nt$]PJ                  PM                  R6R7]PN                  PP                  R87       ]PJ                  PS                  R6R94      R: 4       t*]PJ                  PS                  R6R;4      R< 4       t+]PJ                  PS                  R6R=4      R> 4       t,]PJ                  P[                  R64      R? 4       t.R@ t/RA t0RB t1RC t2RD t3]RE 4       t4RF t5RG t6RH t7RI t8RJ t9RK t:RL t;RM t<RN t=RO t>RP t?RQ t@RR tARS tBRT tCRU tD]! 4       tERV tFRW tGRX tHRY tIRZ tJR[ tK]! R\R]]D]C]! ]P˜                  ]Pš                  4      R^R^R^R^R^R^R^R_7      ]! R`R]]5]3]! ]P˜                  ]Pš                  4      R^R^R^R^R^RaR^R_7      ]! R&R]] ]]! ]P˜                  ]Pš                  4      R^R^R^R^R^Rb7
      ]! R&Rc]!]]! ]P˜                  ]Pš                  4      R^R^R^R^R^Rb7
      ]! R&Rd]"]]! ]P˜                  ]Pš                  4      R^R^R^R^R^Rb7
      ]! ReR]]0]/]! ]P˜                  ]Pš                  4      R^R^R^R^R^RaR^R_7      ]! RfR]]2]1]! ]P˜                  ]Pš                  4      R^R^R^R^R^R^Rg7      ]! RhR]]F]3]! ]P˜                  ]Pš                  4      R^R^R^R^R^RaR^R_7      ]! RiR]]G]3]! ]P˜                  ]Pš                  4      R^R^R^R^R^RaR^R_7      ]! RjR]]?]>]! ]P˜                  ]Pš                  4      R^R^R^R^R^R^Rg7      ]! RkR]]@]>]! ]P˜                  ]Pš                  4      R^R^R^R^R^R^Rg7      ]! RlR]]7]6]! ]P˜                  ]Pš                  4      R^R^R^R^R^R^Rg7      ]! R€RmRnRoR]Rp]Rq]8/]#! ]Pœ                  ]Pž                  4      BRrR^RsR^RtR^RuR^RvR^/B ]! R€RmRwRoR]Rp];Rq]9/]#! ]Pœ                  ]Pž                  4      BRrR^RsR^RtR^RuR^RvR^RxR^RyR^/B ]! R€RmRwRoRzRp];Rq]:/]#! ]Pœ                  ]Pž                  4      BRrR^RsR^RtR^RuR^RvR^RxR^RyR^/B ]! R{R]]=]<]! ]Pž                  4      R^R^R^R^R^R^R^R_7      ]! R€RmR|RoR]Rp]BRq]A/]P                   P£                  4       '       d   ]#! ]Pœ                  ]Pž                  4      MR}]! 4       /BRrR^RsR^RtR^RuR^RvR^/B ]! R~R]]I]H]! ]P˜                  ]Pš                  4      R^R^R^R^R^R^Rg7      ]! R€RmRRoR]Rp]KRq]J/]#! ]Pž                  4      BRrR^RsR^RtR^RuR^RvR^RxR^/B .tRR# )�é    N©Úmap)Úflex_attention)Ú	flex_gemm)Ú_create_empty_block_maskÚcreate_block_maskr   )Úmake_tensor)Úinline_asm_elementwise)Úall_types_andÚcustom_typesÚempty_types)ÚOpInfoÚSampleInput)Úmark_compile_region)ÚInvokeQuantÚinvoke_quant_packed©Úregister_hook_opc              +   ó¾   "  € \         P                  ! \        WVR 7      p\        V! ^^^R^R7      V! ^^^R^R7      .V! ^R^R7      V! ^R^R7      3R7      x € R# 5i)©ÚdeviceÚdtypeÚrequires_gradçš™™™™™¹?©ÚlowÚhigh©ÚargsN©Ú	functoolsÚpartialr	   r   ©Úopinfor   r   r   ÚkwargsÚmake_args   &&&&, Ú_/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/testing/_internal/hop_db.pyÚsample_inputs_mapr(      sg   é € Ü× Ò Ü˜F¸}ô€Hô Ù	�!�Q˜˜s¨Ô	+©X°a¸¸AÀ3ÈQÔ-OÐPÙ�q˜c¨Ô*©H°Q¸CÀaÔ,HÐIôô ùs   ‚AAc                 óü   € V ^ ,          P                  4       P                  R4      V,          V ^,          VP                  4       ,           P                  4       P	                  V ^,          P                  4       4      .# )r   ç      ð?)ÚcosÚadd_ÚsinÚcos_ÚviewÚsize©ÚxÚy0Úy1s   &&&r'   Úinner_fr5   #   sQ   € Øˆa�D�H‰H‹J�O‰O˜CÓ  2Õ%¨¨!­¨r¯v©v«x­×'=Ñ'=Ó'?×'DÑ'DÀQÀqÅTÇYÁYÃ[Ó'QÐRÐRó    c                 ó    € R  p\        W0W4      # )c                 ó   € \        WV4      # ©N©r5   r1   s   &&&r'   ÚfÚsimple_map.<locals>.f(   s   € Ü�q˜bÓ!Ð!r6   r   )Úxsr3   r4   r;   s   &&& r'   Ú
simple_mapr>   '   s   € ò"ô ˆq�bÓÐr6   c                 ó    € R  p\        W0W4      # )c                 ó    € R  p\        W0W4      # )c                 ó   € \        WV4      # r9   r:   r1   s   &&&r'   Úf2Ú"nested_map.<locals>.f1.<locals>.f20   s   € Ü˜1 "Ó%Ð%r6   r   ©Úxxr3   r4   rB   s   &&& r'   Úf1Únested_map.<locals>.f1/   s   € ò	&ô �2˜2Ó"Ð"r6   r   ©r=   r3   r4   rF   s   &&& r'   Ú
nested_maprI   .   s   € ò#ô ˆr�rÓÐr6   c                 ó    € R  p\        W0W4      # )c                 ó    € R  p\        W0W4      # )c                 ó    € R  p\        W0W4      # )c                 ó   € \        WV4      # r9   r:   r1   s   &&&r'   rB   Ú5triple_nested_map.<locals>.f0.<locals>.f1.<locals>.f2;   s   € Ü˜q bÓ)Ð)r6   r   rD   s   &&& r'   rF   Ú)triple_nested_map.<locals>.f0.<locals>.f1:   s   € ò*ô �r˜rÓ&Ð&r6   r   rH   s   &&& r'   Úf0Útriple_nested_map.<locals>.f09   s   € ò	'ô �2˜2Ó"Ð"r6   r   )r=   r3   r4   rP   s   &&& r'   Útriple_nested_maprR   8   s   € ò#ô ˆr�rÓÐr6   c                  ó@   € R \        4       R\        V !  R\        4       /# )ÚdtypesÚdtypesIfCUDAÚdtypesIfXPU)r   r   )rT   s   *r'   Úcuda_only_dtypesrW   E   s$   € à”+“-Øœ fÑ-Ø”{“}ðð r6   Úcustom_function_callÚautograd_function_applyÚrun_and_save_rng_stateÚrun_with_rng_stateÚrun_dtensor_rng_opÚgraphsafe_run_with_rng_stateÚ	out_dtypeÚtrace_wrappedÚtag_activation_checkpointÚexecutorch_call_delegateÚwrapÚwrap_with_set_grad_enabledÚauto_functionalized_v2Úassociative_scanÚ
flat_applyÚwrap_with_autocastÚwrap_activation_checkpointÚrun_const_graphÚauto_functionalizedr   Úmap_implÚwith_effectsÚstrict_modeÚ_export_tracepointÚcall_torchbindÚtriton_kernel_wrapper_mutationÚ triton_kernel_wrapper_functionalÚhints_wrapperÚdynamo_bypassing_wrapperÚforeach_mapÚaoti_call_delegateÚprintÚinductor_compiled_codeÚinvoke_leaf_functionÚcontrol_depsztestlib::mutating_custom_opz8(Tensor(a!) x, Tensor(b!) z) -> (Tensor, Tensor, Tensor))ÚtagsÚcpuc                 ó”   € V P                  ^4       VP                  ^4       V P                  4       VP                  4       W,           3# ©é   ©r,   Úclone©r2   Úzs   &&r'   Úfoo_impl_cpurƒ   ˆ   ó3   € à‡F�Fˆ1„IØ‡F�Fˆ1„IØ�7‰7‹9�a—g‘g“i ¥Ð&Ð&r6   Úcudac                 ó”   € V P                  ^4       VP                  ^4       V P                  4       VP                  4       W,           3# r}   r   r�   s   &&r'   Úfoo_impl_cudar‡   �   r„   r6   Úxpuc                 ó”   € V P                  ^4       VP                  ^4       V P                  4       VP                  4       W,           3# r}   r   r�   s   &&r'   Úfoo_impl_xpurŠ   –   r„   r6   c                 óP   € V P                  4       VP                  4       W,           3# r9   ©r€   r�   s   &&r'   Úfoo_impl_abstractr�   �   s   € à�7‰7‹9�a—g‘g“i ¥Ð&Ð&r6   c           
   +   óx   "  € \         P                  ! \        WVR 7      p\        V! ^^^R^R7      4      x € R# 5i©r   r   r   Nr    r#   s   &&&&, r'   Úsample_inputs_condr�   ¢   ó6   é € Ü× Ò Ü˜F¸}ô€Hô ‘h˜q ! Q¨C°aÔ8Ó
9Ô9ùó   ‚8:c                 ó\   € \         P                  ! V P                  4       ^8„  R R V .4      # )é   c                 ó$   € V P                  4       3# r9   ©r+   ©r2   s   &r'   Ú<lambda>Úsimple_cond.<locals>.<lambda>ª   s   € ¨a¯e©e«g©Zr6   c                 ó$   € V P                  4       3# r9   ©r-   r—   s   &r'   r˜   r™   ª   s   € ÀAÇEÁEÃGÁ:r6   )ÚtorchÚcondÚsumr—   s   &r'   Úsimple_condrŸ   ©   s&   € Ü�:Š:�a—e‘e“g ‘kÑ#7Ñ9MÐPQÈsÓSÐSr6   c           
   +   óx   "  € \         P                  ! \        WVR 7      p\        V! ^^^R^R7      4      x € R# 5ir�   r    r#   s   &&&&, r'   Úsample_inputs_switchr¡   ­   r‘   r’   c                 ór   € ^ RI Hp V P                  4       ^8„  P                  4       pV! VR R R 3V .4      # )r   )Úswitchc                 ó$   € V P                  4       3# r9   r–   r—   s   &r'   r˜   Úsimple_switch.<locals>.<lambda>¼   s   € �A—E‘E“G‘:r6   c                 ó$   € V P                  4       3# r9   r›   r—   s   &r'   r˜   r¥   ¼   s   € ¨!¯%©%«'©r6   c                 ó$   € V P                  4       3# r9   )Úabsr—   s   &r'   r˜   r¥   ¼   s   € ÀÇÁÃ¹zr6   )Útorch._higher_order_ops.switchr£   rž   Úlong)r2   r£   Úidxs   &  r'   Úsimple_switchr¬   ´   s@   € õ 6à�5‰5‹7�Q‰;×
Ñ
Ó
€CÙØÙ	Ñ3Ñ5IÐJØ	
ˆóð r6   c           
   +   óx   "  € \         P                  ! \        WVR 7      p\        V! ^^^R^R7      4      x € R# 5ir�   r    r#   s   &&&&, r'   Úsample_inputs_invoke_subgraphr®   Á   r‘   r’   c                 ó.   € \         P                  ! V 4      # r9   ©rœ   r-   r—   s   &r'   Úfn_for_invoke_subgraphr±   È   s   € ä�9Š9�Q‹<Ðr6   c                 ó   € \        V 4      # r9   )r±   r—   s   &r'   Úsimple_invoke_subgraphr³   Í   s   € Ü! !Ó$Ð$r6   c              +   ó�   "  € \         P                  ! \        WR R7      p\        V! ^^^R^R7      V! ^^^R^R7      4      x € R# 5i©Fr   r   r   Nr    r#   s   &&&&, r'   Ú sample_inputs_auto_functionalizer¶   Ñ   sJ   é € Ü× Ò Ü˜F¸uô€Hô Ù��A�q˜c¨Ô*©H°Q¸¸1À#ÈAÔ,Nóô ùó   ‚AAc                 óT   € \         P                  P                  P                  W4      # r9   )rœ   ÚopsÚtestlibÚmutating_custom_opr�   s   &&r'   Úsimple_auto_functionalizer¼   Ú   s   € Ü�9‰9×Ñ×/Ñ/°Ó5Ð5r6   c              +   ó´   a
"  € \         P                  ! \        WVR 7      o
R pV
3R l\        ^4       4       w  rgp\	        Wg4      p	\        WgW…V	4      x € R# 5i)r   c                 ó   € W,           # r9   © ©ÚscoreÚbÚhÚmÚns   &&&&&r'   Ú	score_modÚ/sample_inputs_flex_attention.<locals>.score_modã   s
   € Ø�yÐr6   c           
   3   ó@   <"  € T F  pS! ^^^€^R^R7      x € K  	  R# 5i©r”   r   r   Nr¿   ©Ú.0Ú_r&   s   & €r'   Ú	<genexpr>Ú/sample_inputs_flex_attention.<locals>.<genexpr>æ   s#   øé € ÐIÁ¸1‰x˜˜1˜c 1¨#°A×6Ñ6Ãùó   ƒN)r!   r"   r	   Úranger   r   )r$   r   r   r   r%   rÆ   ÚqÚkÚvÚ
block_maskr&   s   &&&&,     @r'   Úsample_inputs_flex_attentionrÕ   Þ   sP   øé € Ü× Ò Ü˜F¸}ô€Hòô JÄÀaÄÓI�G€Aˆ!Ü)¨!Ó/€JÜ
�a˜A¨*Ó
5Ô5ùs   ƒAAc              +   óØ  a"  € \         P                  ! \        WR R7      oR pR pV3R l\        ^4       4       w  rxp	\	        V^^^€^€VR7      p
RVP                  R
4      R,          ,          p\        WxW•V
P                  4       V/ 4      w  rÍp\        VW‰VP                  4       VP                  4       \        P                  ! V4      RVRV
P                  4       V/ RR3R	7      x € R# 5i)Fr   c                 ó   € V # r9   r¿   rÀ   s   &&&&&r'   rÆ   Ú8sample_inputs_flex_attention_backward.<locals>.score_modò   ó   € Øˆr6   c                 ó
   € W#8¬  # r9   r¿   )rÂ   rÃ   rÄ   rÅ   s   &&&&r'   Úmask_modÚ7sample_inputs_flex_attention_backward.<locals>.mask_modõ   s	   € Ø‰vˆr6   c           
   3   ó@   <"  € T F  pS! ^^^€^R^R7      x € K  	  R# 5irÉ   r¿   rÊ   s   & €r'   rÍ   Ú8sample_inputs_flex_attention_backward.<locals>.<genexpr>ø   ó#   øé € ÐJÁ¸A‰x˜˜1˜c 2¨3°Q×7Ñ7ÃùrÏ   ©ÚBÚHÚQ_LENÚKV_LENr   r*   ç      à?Nr   éÿÿÿÿr¿   )r!   r"   r	   rÐ   r   r0   Úflex_attention_hopÚas_tupler   Údetachrœ   Ú	rand_like)r$   r   r   r   r%   rÆ   rÛ   rÑ   rÒ   rÓ   rÔ   ÚscaleÚoutÚ	logsumexprÌ   r&   s   &&&&,          @r'   Ú%sample_inputs_flex_attention_backwardrî   ë   sà   øé € ô × Ò Ü˜F¸uô€Hòòô KÄÀqÄÓJ�G€Aˆ!Ü" 8¨q°A¸SÈÐU[Ô\€JØ�!—&‘&˜“* Õ#Õ#€EÜ*Ø	ˆa˜J×/Ñ/Ó1°5¸"óÑ€C�Aô Ø	à�#—*‘*“, 	× 0Ñ 0Ó 2´E·O²OÀCÓ4HÈ$Ø�t˜Z×0Ñ0Ó2Ø�2�r˜2ð
ôô ùs   ƒC'C*c              +   ó0  aa"  € \         P                  ! \        WR R7      o\        P                  ! R^€V\        P
                  R7      oR pV3R lpV3R l\        ^4       4       w  rxp	\        V^^^€^€VR7      p
RVP                  R4      R,          ,          p\        WxW•V
P                  4       V/ 4      w  rÍp\        VW‰VP                  4       VP                  4       \        P                  ! V4      R	VR	V
P                  4       V/ RR3R
7      x € R	# 5i)Fr   )r   r   c                 ó   € V # r9   r¿   rÀ   s   &&&&&r'   rÆ   ÚIsample_inputs_flex_attention_backward_explicit_buffers.<locals>.score_mod  rÙ   r6   c                 ó   <€ VS,           V8¬  # r9   r¿   )rÂ   rÃ   rÄ   rÅ   Úmask_offsets   &&&&€r'   rÛ   ÚHsample_inputs_flex_attention_backward_explicit_buffers.<locals>.mask_mod  s   ø€ Ø�;� !Ñ#Ð#r6   c           
   3   ó@   <"  € T F  pS! ^^^€^R^R7      x € K  	  R# 5irÉ   r¿   rÊ   s   & €r'   rÍ   ÚIsample_inputs_flex_attention_backward_explicit_buffers.<locals>.<genexpr>  rß   rÏ   rà   r*   rå   Nr   r¿   ræ   )r!   r"   r	   rœ   ÚfullÚint32rÐ   r   r0   rç   rè   r   ré   rê   )r$   r   r   r   r%   rÆ   rÛ   rÑ   rÒ   rÓ   rÔ   rë   rì   rí   rÌ   r&   ró   s   &&&&,          @@r'   Ú6sample_inputs_flex_attention_backward_explicit_buffersrù     sú   ùé € ô × Ò Ü˜F¸uô€Hô —*’*˜R ¨V¼5¿;¹;ÔG€Kòõ$ô KÄÀqÄÓJ�G€Aˆ!Ü" 8¨q°A¸SÈÐU[Ô\€JØ�!—&‘&˜“* Õ#Õ#€EÜ*Ø	ˆa˜J×/Ñ/Ó1°5¸"óÑ€C�Aô Ø	à�#—*‘*“, 	× 0Ñ 0Ó 2´E·O²OÀCÓ4HÈ$Ø�t˜Z×0Ñ0Ó2Ø�2�r˜2ð
ôô ùs   „DDc                 ón   € \         P                  P                  P                  V VVVVVVVVV	V
VVV4      # r9   )rœ   r¹   Úhigher_orderÚflex_attention_backward)ÚqueryÚkeyÚvaluerì   rí   Úgrad_outÚgrad_logsumexpÚfw_graphÚjoint_graphrÔ   rë   Úkernel_optionsÚscore_mod_other_buffersÚmask_mod_other_bufferss   &&&&&&&&&&&&&&r'   Úsimple_flex_attention_backwardr  &  sK   € ô  �9‰9×!Ñ!×9Ñ9ØØØØØØØØØØØØØØóð r6   c           
   +   óŒ   "  € \         P                  ! \        WR R7      p\        V! ^^R^R7      V! ^^R^R7      4      x € R# 5irµ   r    r#   s   &&&&, r'   Úsample_inputs_flex_gemmr	  H  sB   é € Ü× Ò Ü˜F¸uô€Hô ‘h˜q !¨°1Ô5±xÀÀ1È#ÐTUÔ7VÓ
WÔWùs   ‚AAc                 ó@   € R  p\        \        P                  W3V4      # )c                 ó0   € V ^,           P                  4       # ©é   )Úrelu)Úaccs   &r'   Úepilogue_fnÚ%simple_flex_gemm.<locals>.epilogue_fnP  s   € Ø�a•�~‰~ÓÐr6   )r   rœ   Úmm)ÚarÂ   r  s   && r'   Úsimple_flex_gemmr  O  s   € ò ô ”U—X‘X ˜v {Ó3Ð3r6   c              +   ó¢   "  € \         P                  ! \        WR R7      p\        \        P
                  ! ^4      V! ^^^R^R7      4      x € R# 5irµ   )r!   r"   r	   r   rœ   Útensorr#   s   &&&&, r'   Úsample_inputs_while_loopr  V  sF   é € Ü× Ò Ü˜F¸uô€Hô Ü�Š�Q‹Ù��A�q˜c¨Ô*óô ùs   ‚AAc                 óP   € R  pR p\         P                  P                  W#W34      # )c                 ó   € V ^ 8„  # ©r   r¿   ©Úiter_tr2   s   &&r'   Úcond_fnÚ"simple_while_loop.<locals>.cond_fna  ó   € Ø˜‰zÐr6   c                 ó4   € V ^,
          VP                  4       3# r  r–   r  s   &&r'   Úbody_fnÚ"simple_while_loop.<locals>.body_fnd  ó   € Ø˜�z˜1Ÿ5™5›7Ð"Ð"r6   )rœ   Ú_higher_order_opsÚ
while_loop©r  r2   r  r!  s   &&  r'   Úsimple_while_loopr'  `  s)   € òò#ô ×"Ñ"×-Ñ-¨gÀÀÓLÐLr6   c                 ób   € R  pR p\         P                  P                  W#W3\        4       4      # )c                 ó   € V ^ 8„  # r  r¿   r  s   &&r'   r  Ú/simple_while_loop_stack_output.<locals>.cond_fnk  r  r6   c                 ó4   € V ^,
          VP                  4       3# r  r–   r  s   &&r'   r!  Ú/simple_while_loop_stack_output.<locals>.body_fnn  r#  r6   )rœ   r$  Úwhile_loop_stack_outputÚtupler&  s   &&  r'   Úsimple_while_loop_stack_outputr/  j  s2   € òò#ô ×"Ñ"×:Ñ:Ø˜6˜+¤u£wóð r6   c              +   ó�   "  € \         P                  ! \        WR R7      p\        V! ^^^R^R7      V! ^^^R^R7      4      x € R# 5irµ   r    r#   s   &&&&, r'   Úsample_inputs_local_map_hopr1  v  sK   é € ä× Ò Ü˜F¸uô€Hô Ù��A�q˜c¨Ô*Ù��A�q˜c¨Ô*óô ùr·   c                 ó`  € R  p\         P                  P                  V4      p\         P                  P	                  4       '       g   \        R4      h^ RIHp RV! 4       V! 4       V! 4       3RV! 4       V! 4       V! 4       33/VP                  R&   \         P                  P                  W0V4      # )c                 óL   € V P                  4       VP                  4       ,           # r9   )r+   r-   )Úinp1Úinp2s   &&r'   Úbody_gmÚ%simple_local_map_hop.<locals>.body_gm‚  s   € Ø�x‰x‹z˜DŸH™H›JÕ&Ð&r6   z*Expected torch.distributed to be available)Ú	ReplicateÚin_placementsÚout_placementsÚlocal_map_kwargs)rœ   ÚfxÚsymbolic_traceÚdistributedÚis_availableÚAssertionErrorÚ(torch.distributed.tensor.placement_typesr8  Úmetar$  Úlocal_map_hop)r4  r5  r6  Úgmr8  s   &&   r'   Úsimple_local_map_hoprE  �  s–   € ò'ô 
�‰×	 Ñ	  Ó	)€Bä×Ñ×)Ñ)×+Ò+ÜÐIÓJÐJÝBð 	™)›+¡y£{±I³KÐ@Ø™I›K©«±i³kÐBÐDð#€B‡G�GÐÑô ×"Ñ"×0Ñ0°¸4Ó@Ð@r6   c              +   óŽ   "  € \         P                  ! \        WVR 7      p\        V! ^^R^R7      V! ^^^R^R7      4      x € R# 5ir�   r    r#   s   &&&&, r'   Úsample_inputs_scanrG  ”  sI   é € Ü× Ò Ü˜F¸}ô€Hô Ù��A˜3 QÔ'Ù��A�q˜c¨Ô*óô ùs   ‚AAc                 óH   € R  p\         P                  P                  W V4      # )c                 óB   € W,          V,           pW P                  4       3# r9   rŒ   )Úcarryr2   Úresults   && r'   Ú
combine_fnÚsimple_scan.<locals>.combine_fnŸ  s   € Ø•˜Q•ˆØ—{‘{“}Ð$Ð$r6   )rœ   r$  Úscan)Úinitr=   rL  s   && r'   Úsimple_scanrP  ž  s"   € ò%ô ×"Ñ"×'Ñ'¨
¸"Ó=Ð=r6   c                 ó<   € R  p\        WV 4      ^ ,          R,          # )c                 ó>   € \         P                  ! V 4      V,          3# r9   r°   )r2   Úys   &&r'   ÚfnÚsimple_invoke_quant.<locals>.fnª  s   € Ü—	’	˜!“˜qÕ Ð"Ð"r6   ç       @)Úquant_tracer©r2   rT  s   & r'   Úsimple_invoke_quantrY  ©  s    € ò#ô ˜˜qÓ! !Õ$ sÕ*Ð*r6   c                 ó:   € R  p\        W4      ^ ,          R,          # )c                 ó0   € \         P                  ! V 4      3# r9   r°   r—   s   &r'   rT  Ú&simple_invoke_quant_packed.<locals>.fn±  s   € Ü—	’	˜!“ˆÐr6   rV  )r   rX  s   & r'   Úsimple_invoke_quant_packedr]  °  s   € òô ˜rÓ% aÕ(¨3Õ.Ð.r6   c           
   +   óx   "  € \         P                  ! \        WR R7      p\        V! ^^^R^R7      4      x € R# 5i)Tr   r   r   Nr    r#   s   &&&&, r'   Úsample_inputs_register_hookr_  ·  s6   é € Ü× Ò Ü˜F¸tô€Hô ‘h˜q ! Q¨C°aÔ8Ó
9Ô9ùr’   c                 ó   € R  p\        W4      # )c                 ó   € V ^,          # )r”   r¿   )Úgrads   &r'   ÚhookÚ"simple_register_hook.<locals>.hook¿  s   € Ø�a�xˆr6   r   )r2   rc  s   & r'   Úsimple_register_hookre  ¾  s   € òô ˜AÓ$Ð$r6   c           
   +   óx   "  € \         P                  ! \        WVR 7      p\        V! ^^^R^R7      4      x € R# 5ir�   r    r#   s   &&&&, r'   Úsample_inputs_inline_asmrg  Å  r‘   r’   c                 ó¶   € \         P                  P                  '       d   \        V R R\         P                  R7      # \        V RR\         P                  R7      # )zv_mov_b32_e32 $0, $1z=v, v)Úasm_strÚconstraintsr   zmov.f32 $0, $1;z=f,f)rœ   ÚversionÚhipr
   Úfloat32r—   s   &r'   Úsimple_inline_asmrn  Ì  sL   € Ü‡}�}××ÐÜ%ØØ*ØÜ—-‘-ô	
ð 	
ô "Ø	Ð$°&ÄÇÁôð r6   rN  ÚsimpleF)ÚnameÚvariant_test_nameÚopÚsample_inputs_funcrT   Úsupports_outÚcheck_batched_gradÚcheck_batched_gradgradÚcheck_batched_forward_gradÚ"check_inplace_batched_forward_gradÚsupports_autogradÚsupports_gradgradÚinvoke_subgraphT)
rp  rq  rr  rs  rT   rt  ru  rv  rw  rx  ÚnestedÚtriple_nestedr�   r£   )rp  rq  rr  rs  rT   rt  ru  rv  rw  rx  ry  Úinvoke_quantr   r%  r-  Úauto_functionalizerp  r   rq  rr  rs  rt  ru  rv  rw  rx  rü   ry  rz  Úexplicit_buffersr   rC  rT   Úregister_hookr
   r¿   )Sr!   rœ   Ú#functorch.experimental.control_flowr   Ú&torch._higher_order_ops.flex_attentionr   rç   Ú!torch._higher_order_ops.flex_gemmr   Ú!torch.nn.attention.flex_attentionr   r   Útorch.testingr	   Ú.torch._higher_order_ops.inline_asm_elementwiser
   Ú$torch.testing._internal.common_dtyper   r   r   Ú#torch.testing._internal.opinfo.corer   r   Ú'torch._higher_order_ops.invoke_subgraphr   Útorch._higher_order_opsr   r   Ú%torch._higher_order_ops.register_hookr   r(   r5   r>   rI   rR   rW   Ú0FIXME_hop_that_doesnt_have_opinfo_test_allowlistÚlibraryÚdefineÚTagÚpt2_compliant_tagÚimplrƒ   r‡   rŠ   Úregister_faker�   r�   rŸ   r¡   r¬   r®   r±   r³   r¶   r¼   rÕ   rî   rù   r  r	  r  r  r'  r/  r1  rE  rG  rP  rW  rY  r]  r_  re  rg  rn  ÚboolÚhalfÚfloat16rm  r>  r?  Úhop_dbr¿   r6   r'   Ú<module>r˜     s	  ðó ã Ý 3õõ 8÷ñ õ
 &Ý Qß YÑ Yß CÝ Gß DÝ BòòSòòò
òð,$4Øð$4àð$4ð ð$4ð ð	$4ð
 ð$4ð #ð$4ð ð$4ð ð$4ð  ð$4ð ð$4ð ð$4ð !ð$4ð ð$4ð ð$4ð ð$4ð  ð!$4ð" !ð#$4ð$ ð%$4ð& ð'$4ð( 
ð)$4ð* ð+$4ð, ð-$4ð. ð/$4ð0 ð1$4ð2 ð3$4ð4 %ð5$4ð6 'ð7$4ð8 ð9$4ð: ð;$4ð< ð=$4ð> ð?$4ð@ ðA$4ðB ðC$4ðD ðE$4ðF ðG$4Ð 0ðL ‡�× Ñ Ø!Ø>Ø	�‰×	$Ñ	$ð ô ð ‡�×ÑÐ1°5Ó9ñ'ó :ð'ð ‡�×ÑÐ1°6Ó:ñ'ó ;ð'ð ‡�×ÑÐ1°5Ó9ñ'ó :ð'ð ‡�×ÑÐ:Ó;ñ'ó <ð'ò:òTò:ò
ò:ð ñó ðò%òò6ò
6òò:ò<òDXò4òòMò	òòAò&ò>ñ ‹}€ò+ò/ò:ò%ò:òñ ØØ"ØØ-Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øàôñ ØØ"Ø!Ø8Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øàôñ ØØ"ØØ,Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0ôñ ØØ"ØØ,Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0ôñ ØØ)ØØ,Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0ôñ ØØ"ØØ-Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øàôñ ØØ"ØØ/Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0ð  ôñ  ØØ"ØØ8Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øàôñ Ø"Ø"Ø%Ø8Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øàôñ ØØ"ØØ3Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øôñ Ø&Ø"Ø)Ø3Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øôñ Ø!Ø"Ø$Ø;Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øôñ ñ Øðà"ðð ðð 8ð	ñ
 ˜5Ÿ=™=¨%¯-©-Ó
8ñð ðð !ðð  %ðð $)ðð ,1òñ ñ Ø&ðà"ðð *ðð Að	ñ
 ˜5Ÿ=™=¨%¯-©-Ó
8ñð ðð !ðð  %ðð $)ðð ,1ðð  ðð  òñ ñ Ø&ðà,ðð *ðð Rð	ñ
 ˜5Ÿ=™=¨%¯-©-Ó
8ñð ðð !ðð  %ðð $)ðð ,1ðð  ðð  òñ ØØ"ØØ2Ù˜EŸM™MÓ*ØØ Ø$Ø#(Ø+0ØØôñ ñ Øðà"ðð  ðð 7ð	ð × Ñ ×-Ñ-×/Ò/ñ ˜UŸ]™]¨E¯M©MÔ:à™K›MÐ*ñð ðð !ðð  %ðð $)ðð ,1òñ  ØØ"ØØ6Ù˜UŸZ™Z¨¯©Ó4ØØ Ø$Ø#(Ø+0Øôñ ñ Ø%ðà"ðð ðð 4ð	ñ
 ˜5Ÿ=™=Ó
)ñð ðð !ðð  %ðð $)ðð ,1ðð  òðuG
‚r6   