
    3jfK                     H   % S r SSKrSSKrSSKrSSKrSSKrSSKrSSKrSSK	r	SSK
r
SSKrSSKJrJrJr  SSKJrJrJr  SSKJr  SSKJr  SSKrSSKJr  SSKJr  SS	KJr  S
SKJrJ r J!r!J"r"  S
SK#J$r$J%r%J&r&J'r'  S
SK(J)r)J*r*J+r+  S
SK
J,r,J-r-J.r.  S
SK/J0r0J1r1  Sr2\
Rf                  S-  \4S'    SSK5r2\ Rn                  r7Sr8\Rr                  " \:5      r;\" S5      r<S\Rz                  S-  S\Rz                  S-  4S jr>S\?S\?4S jr@S rAS rBS\R                  R                  S\S\S\S\E\   4
S jrFS\S\G4S  jrH\S\Rz                  S\Rz                  4S! j5       rI\S\E\   \J\S"4   -  \K\\4   -  S\L4S# j5       rIS\S\Rz                  \L-  4S$ jrIS\?4S% jrMSpS\?S&\
R                  S'\?SS4S( jjrOSS).S*\-S+\PS,\S-\S.\PS\,4S/ jjrQ " S0 S15      rR " S2 S35      rS " S4 S55      rT " S6 S75      rU " S8 S95      rVS&\?S\?4S: jrWS&\?S\?4S; jrXS&\?S\?4S< jrYS=\?S\?4S> jrZS&\?S\?4S? jr[   SqS@\SA\S"\4   SB\PSC\PS-  SD\PS-  SE\PSS4SF jjr\SG\R                  S\E\Rz                     S\S"\4   4SH jr^  SrSI\LSJ\LSK\GSL\LS\?4
SM jjr_\R                  SNS4SO\\P   SP\\P   SQ\R                  SR\?\R                  -  SS\PS\Rz                  4ST jjrc\" SU5      rdS\G4SV jreSA\\<\d4   SW\S\\<\d4   4SX jrfSSY SZ.S[\gS\\?S]\?SW\S^\?S-  S_\\S"\4   /\S"\4   4   S\g4S` jjrhSA\\<\d4   S\\<\d4   4Sa jriSA\\<\d4   S\\<\d4   4Sb jrjSA\\<\d4   S\\<\d4   4Sc jrkSA\\<\d4   S\\<\d4   4Sd jrlSA\\<\d4   S\\<\d4   4Se jrmSA\\<\d4   S\\<\d4   4Sf jrnSA\\<\d4   S\\<\d4   4Sg jroSA\\<\d4   S\\<\d4   4Sh jrpSsSi\GSS4Sj jjrqSk\\<\d4   Sl\<R                  Sm\<R                  S\d4Sn jrtSqu\R                  S\\E\R2                  R                        4So j5       rwg! \6 a    Sr2 GNqf = f)ta1  Testing utilities and infrastructure for Dynamo.

This module provides a comprehensive set of testing utilities including:
- Test result collection and validation
- Graph manipulation and comparison tools
- Test case management and execution helpers
- Specialized test decorators for different Python versions and features
- RNG state management
- Compilation counting and monitoring
- Debug utilities for bytecode transformation

The utilities in this module are used across Dynamo's test suite to ensure
consistent testing patterns and proper test isolation.
    N)Callable	GeneratorSequence)AnyoverloadTypeVar)	ParamSpec)patch)fx)	aot_eager)OutputGraph   )config
eval_frameoptimize_assertreset)create_instructiondebug_checksis_generatortransform_code_object)CheckFunctionManager	CompileIdGuardedCode)ConvertFrameReturnDynamoFrameTypewrap_guarded_code)CompileCounterIntsamenp   _Pxreturnc                 x    U c  g U R                  5       R                  5       R                  U R                  5      $ N)detachclonerequires_grad_requires_gradr"   s    O/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/_dynamo/testing.pyclone_mer,   ?   s.    y88:,,Q__==    namec                 2    [         R                  " SSU 5      $ )Nz^_orig_mod[.] resub)r.   s    r+   remove_optimized_module_prefixr4   E       66"B--r-   c                    ^^^ SSK Jm  S mS mUUU4S jn[        R                  " USS9" U 5      " U0 UD6  TR                  T4$ )Nr   )InstructionTranslatorc                 T   > U mTR                  5       R                  R                  mU $ r%   )
current_txoutputregion_tracker)_gmargskwargsr7   gmr;   s      r+   extract_graph_backend8extract_graph_and_tracker.<locals>.extract_graph_backendO   s)     .99;BBQQ
r-   T)backend	fullgraph)torch._dynamo.symbolic_convertr7   torchcompilegraph)fnr=   r>   r@   r7   r?   r;   s       @@@r+   extract_graph_and_trackerrI   I   sD    D	BN 
MM/4@DdUfU88^##r-   c                     [        5       n[        R                  " US9" U 5      " U0 UD6nXCR                  UR                  UR
                  4$ )N)rB   )AotEagerAndRecordGraphsrE   rF   graphs	fw_graphs	bw_graphs)rH   r=   r>   rB   results        r+   extract_graphrP   Z   sD    %'G]]7+B/@@F>>7#4#4g6G6GGGr-   model
predictionlossexample_inputsc                 @   / nUR                  U5        UR                  U5        0 n0 nU R                  5        Hk  u  px[        U [        R                  5      (       a  [        U5      nUn	UR                  n
UR                  c  [        R                  " U5      n
XUS-   '   XU'   Mm     UR                  U5        UR                  U5        0 nU R                  5        H3  u  p|[        U [        R                  5      (       a  [        U5      nXU'   M5     UR                  U5        U Ht  n[        U[        [        45      (       a  UR                  S U 5       5        M8  [        U[        R                  5      (       d  MY  UR                  UR                  5        Mv     U$ )Nz.gradc              3   |   #    U  H2  n[        U[        R                  5      (       d  M$  UR                  v   M4     g 7fr%   )
isinstancerE   Tensorgrad).0inps     r+   	<genexpr>"collect_results.<locals>.<genexpr>   s#     Xw*S%,,:W8388ws   #<<)appendnamed_parametersrW   r   OptimizedModuler4   rY   rE   
zeros_likenamed_bufferstuplelistextendrX   )rQ   rR   rS   rT   resultsgradsparamsr.   param
param_copyrY   buffersbufferexamples                 r+   collect_resultsrn   `   sU    GNN:NN4 EF--/eZ77881$7D
zz::##E*D $dWn!t 0 NN5NN6G++-eZ77881$7D . NN7!gt}--NNXwXX'5<<00w||, " Nr-   outc                    [        U [        R                  5      (       a  U R                  $ [        U [        [
        45      (       a  [        S U  5       5      $ U c  g[        U [        5      (       a  g[        S[        U 5      5      e)Nc              3   8   #    U  H  n[        U5      v   M     g 7fr%   )requires_bwd_passrZ   r"   s     r+   r\   $requires_bwd_pass.<locals>.<genexpr>   s     5A$Q''   FDon't know how to reduce)
rW   rE   rX   r)   rd   rc   anyintNotImplementedErrortypero   s    r+   rr   rr      sk    #u||$$   	C$	'	'5555		C		
8$s)
DDr-   c                     g r%    r{   s    r+   reduce_to_scalar_lossr~      s    >Ar-   .c                     g r%   r}   r{   s    r+   r~   r~      s     r-   c                 ~   [        U [        R                  5      (       a!  U R                  5       U R	                  5       -  $ [        U [
        [        45      (       a  [        S U  5       5      [        U 5      -  $ [        U 5      R                  S;   a  [        U R                  5      $ [        U 5      R                  S:X  a  U R                  R                  5       $ [        U [        5      (       a:  [        S U R                  5        5       5      [        U R                  5       5      -  $ [!        S[        U 5      5      e)z/Reduce the output of a model to get scalar lossc              3   8   #    U  H  n[        U5      v   M     g 7fr%   r~   rs   s     r+   r\   (reduce_to_scalar_loss.<locals>.<genexpr>   s     9S(++Sru   )MaskedLMOutputSeq2SeqLMOutput!CausalLMOutputWithCrossAttentionsSquashedNormalc              3   8   #    U  H  n[        U5      v   M     g 7fr%   r   )rZ   values     r+   r\   r      s     J\E(//\ru   rv   )rW   rE   rX   sumnumelrd   rc   lenrz   __name__r~   logitsmeandictvalueskeysry   r{   s    r+   r~   r~      s    #u||$$wwy399;&&	C$	'	'9S99CHDD	c		   

 %SZZ00	c		/	/xx||~	C		JSZZ\JJSHHJN
 
 	
 8$s)
DDr-   c                      [         R                  R                  [         R                  R                  [        5      S5      n [         R                  R                  U 5      (       d  [         R                  " U 5        U $ )Nz../debug)ospathjoindirname__file__existsmkdir)r   s    r+   	debug_dirr      sF    77<<1:>D77>>$
Kr-   codeextrac           	      X   [        [        R                  R                  [	        5       U 5      S5       nUR                  [        R                  " U5      R                  5        S[        R                  " U5      R                  5        SU S35        S S S 5        g ! , (       d  f       g = f)Nwz


)	openr   r   r   r   writedisBytecodeinfo)r.   r   r   fds       r+   
debug_dumpr      st    	bggll9;-s	3r
||D!&&()cll4.@.D.D.F-GtE7RTU	
 
4	3	3s   AB
B))skipframe
cache_sizehooks_r   c                   S[         [           S[        SS4S jn[        R                  R                  R                  5       n[        R                  R                  R                  S5         U   [        U R                  5      (       a  [        5       sSSS5        sSSS5        $ [        U R                  5        [        U R                  U5      u  ps[        0 SSS/ SS	0[        5       [        5       U R                  / SS
9n[        [!        U[#        U R                  U5      R$                  ['        S	S	S95      5      sSSS5        sSSS5        $ ! , (       d  f       O= fSSS5        g! , (       d  f       g= f)zused to debug jump updatesinstructionscode_optionsr#   Nc                 p    U R                  S[        S5      5        U R                  S[        S5      5        g )Nr   NOP)insertr   )r   r   s     r+   insert_nops&debug_insert_nops.<locals>.insert_nops   s.    A1%89A1%89r-   debug_insert_nopsF_idr   )r   compiler_fnroot_txexportexport_constraintsframe_statelocal_scopeglobal_scopef_codetorch_function_mode_stackpackage)frame_idframe_compile_id)rd   r   rE   _dynamoutilsget_metrics_contextdynamo_timedr   r   r   r   r   r   localsglobalsr   r   r   guard_managerr   )	r   r   r   r   r   r   metrics_contextr   rG   s	            r+   r   r      s!   
:$s) :3 :4 : mm))==?O				)	)*=	>%%%' AP	>	> 	U\\"'kB!
 <<&(
 !$U\\59GG1q9
+ AP	>	>	>	>	>s1   ,E*/$E	E*&BE=	E*
E	E**
E8c                       \ rS rSrSS jrS\R                  R                  S\\R                     S\
S\4   4S jrSS	 jrS
rg)CompileCounter   r#   Nc                 2    SU l         U R                  5         g Nr   )frame_countclearselfs    r+   __init__CompileCounter.__init__   s    45

r-   r?   rT   .c                     U =R                   S-  sl         UR                  R                   H*  nSUR                  ;   d  M  U =R                  S-  sl        M,     UR
                  $ )Nr   call)r   rG   nodesopop_countforward)r   r?   rT   nodes       r+   __call__CompileCounter.__call__   sM     	AHHNND " # zzr-   c                 l    [         R                  (       a  [        S5      U l        OSU l        SU l        g r   )r   debug_disable_compile_counterr   r   r   r   s    r+   r   CompileCounter.clear   s'    //03D Dr-   )r   r   r#   N)r   
__module____qualname____firstlineno__r   rE   r   GraphModulerd   rX   r   r   r   r   __static_attributes__r}   r-   r+   r   r      sE    ((&&8<U\\8J	#s(	r-   r   c                       \ rS rSrS\SS4S jrS\R                  R                  S\	\R                     S\S\4   4S	 jrSS
 jrSrg)CompileCounterWithBackend   rB   r#   Nc                 L    SU l         Xl        / U l        U R                  5         g r   )r   rB   rL   r   )r   rB   s     r+   r   "CompileCounterWithBackend.__init__  s    4524

r-   r?   rT   .c                 &   SSK Jn  U =R                  S-  sl        UR                  R                   H*  nSUR
                  ;   d  M  U =R                  S-  sl        M,     U R                  R                  U5        U" U R                  5      " X5      $ )Nr   )lookup_backendr   )
backends.registryr   r   rG   r   r   r   rL   r^   rB   )r   r?   rT   r   r   s        r+   r   "CompileCounterWithBackend.__call__  sl     	6AHHNND " # 	2dll+B??r-   c                 z    [         R                  (       a  [        S5      U l        OSU l        SU l        / U l        g r   )r   r   r   r   r   rL   r   s    r+   r   CompileCounterWithBackend.clear  s.    //03D Dr-   )rB   r   rL   r   r   )r   r   r   r   strr   rE   r   r   rd   rX   r   r   r   r   r   r}   r-   r+   r   r      sW      
@((&&
@8<U\\8J
@	#s(	
@r-   r   c                       \ rS rSrS
S jrS\R                  R                  S\\R                     S\
S\4   4S jrS	rg)EagerAndRecordGraphsi  r#   Nc                     / U l         g r%   rL   r   s    r+   r   EagerAndRecordGraphs.__init__  s	    24r-   r?   rT   .c                 P    U R                   R                  U5        UR                  $ r%   )rL   r^   r   )r   r?   rT   s      r+   r   EagerAndRecordGraphs.__call__"  s      	2zzr-   r   r   r   r   r   r   r   rE   r   r   rd   rX   r   r   r   r   r}   r-   r+   r   r     s@    5((&&8<U\\8J	#s(	r-   r   c                       \ rS rSrS
S jrS\R                  R                  S\\R                     S\
S\4   4S jrS	rg)rK   i)  r#   Nc                 .    / U l         / U l        / U l        g r%   )rL   rM   rN   r   s    r+   r    AotEagerAndRecordGraphs.__init__*  s    245757r-   r?   rT   .c                 r  ^  T R                   R                  U5        S[        R                  R                  S[
        [        R                     S[        S[        4   4U 4S jjnS[        R                  R                  S[
        [        R                     S[        S[        4   4U 4S jjn[        UUUUS9$ )Nr?   rT   r#   .c                 R   > TR                   R                  U 5        U R                  $ r%   )rM   r^   r   r?   rT   r   s     r+   fw_compiler5AotEagerAndRecordGraphs.__call__.<locals>.fw_compiler4  !     NN!!"%::r-   c                 R   > TR                   R                  U 5        U R                  $ r%   )rN   r^   r   r  s     r+   bw_compiler5AotEagerAndRecordGraphs.__call__.<locals>.bw_compiler:  r  r-   )r  r  )
rL   r^   rE   r   r   rd   rX   r   r   r   )r   r?   rT   r  r  s   `    r+   r    AotEagerAndRecordGraphs.__call__/  s     	2	$$	6:5<<6H	c3h		$$	6:5<<6H	c3h	 ##	
 	
r-   )rN   rM   rL   r   r   r}   r-   r+   rK   rK   )  s@    8

((&&
8<U\\8J
	#s(	
r-   rK   c                   $    \ rS rSrSS jrS rSrg)InductorAndRecordGraphsiH  Nc                      / U l         / U l        g r%   rL   inductor_graphsr   s    r+   r    InductorAndRecordGraphs.__init__I  s    24;=r-   c                    ^ ^ SS K Js  Jn  T R                  R	                  U5        UR
                  mUU 4S jn[        R                  " USUS9   UR                  X5      sS S S 5        $ ! , (       d  f       g = f)Nr   c                  P   > TR                   R                  U S   5        T" U 0 UD6$ r   )r  r^   )r=   r>   old_compile_fx_innerr   s     r+   patched1InductorAndRecordGraphs.__call__.<locals>.patchedT  s+      ''Q0'888r-   _compile_fx_inner)new)torch._inductor.compile_fx	_inductor
compile_fxrL   r^   r  r
   object)r   r?   rT   compile_fx_modr  r  s   `    @r+   r    InductorAndRecordGraphs.__call__M  sZ    ;;2-??	9 \\.*=7K!,,R@ LKKs   A++
A9r  r   )r   r   r   r   r   r   r   r}   r-   r+   r
  r
  H  s    >Ar-   r
  c                 2    [         R                  " SSU 5      $ )Nz(?m)^ *#.*\n?r0   r1   )r   s    r+   strip_commentr  \  r5   r-   c                     SR                  U R                  S5       Vs/ s H  oR                  5       PM     sn5      $ s  snf )Nr   )r   splitrstrip)r   lines     r+   remove_trailing_spacer"  `  s2    99

40@A0@kkm0@ABBAs   ?c                     U R                  S5      n/ nSnU H6  nUR                  5       S:X  a  U(       a  M   SnOSnUR                  U5        M8     SR                  U5      $ )Nr   Fr0   T)r  stripr^   r   )r   linesrO   	saw_blankr!  s        r+   _squash_blank_linesr'  d  s_    JJtEFI::<2IId  99Vr-   gm_strc                 D    [        U 5      n[        U5      n[        U5      $ r%   )r  r"  r'  )r(  strippedno_trailings      r+   normalize_gmr,  s  s#     V$H'1K{++r-   c                 6    [         R                  " SSU 5      nU$ )z%
Normalize code: remove empty lines.
z[\r\n]+r   r1   )r   normal_codes     r+   empty_line_normalizerr/  {  s     &&T40Kr-   r   rH   nargsexpected_opsexpected_ops_dynamicexpected_frame_countc                    [         R                  (       d  Ub  Un[        5       n[        U5       Vs/ s H  n[        R
                  " SS5      PM     nn[        U5       Vs/ s H  n[        R
                  " SS5      PM     n	nU" U6 n
U" U	6 n[        5         [        U5      " U5      nU" U6 nU" U	6 nU" U6 nU" U	6 n[        5         U R                  [        X5      5        U R                  [        X5      5        U R                  [        X5      5        U R                  [        UU5      5        U R                  UR                  U5        Ub  U R                  UR                  U5        g g s  snf s  snf )N
   )r   assume_static_by_defaultr   rangerE   randnr   r   
assertTruer   assertEqualr   r   )r   rH   r0  r1  r2  r3  actualr   args1args2correct1correct2opt_fnval1aval2aval1bval2bs                    r+   standard_testrE    s9    **/C/O+F*/,7,QU[[R ,E7*/,7,QU[[R ,E75zH5zH	GV$R(FENEENEENEENE	GOOD)*OOD)*OOD)*OOD)*V'')=>,7  ! 87s   !E."!E3r?   c                     U R                   $ r%   )r   )r?   rT   s     r+   dummy_fx_compilerG    s     ::r-   speeduppvalue
is_correctpvalue_thresholdc                 :    U(       d  gX:  a  U S S3$ U S SUS 3$ )NERRORz.3fzx SAMEzx p=z.2fr}   )rH  rI  rJ  rK  s       r+   format_speeduprN    s4      #f%%c]$vcl++r-   cpusizestridedtypedevice
extra_sizec                 V   Un[        S U  5       5      (       a!  U[        S [        X5       5       5      S-   -  nUR                  (       a  U[        R
                  :X  aE  [        R                  " SSU4[        R                  US9R                  [        R
                  5      nOoUR                  S:X  a3   [        R                  " U[        R                  US9R                  US9nO,[        R                  " XRUS9nO[        R                  " U/X#S9n[        R                  " X`U5      $ )	Nc              3   *   #    U  H	  oS :  v   M     g7f)r   Nr}   )rZ   ss     r+   r\   rand_strided.<locals>.<genexpr>  s     
$Qq5$s   c              3   6   #    U  H  u  pUS -
  U-  v   M     g7f)r   Nr}   )rZ   shaperQ  s      r+   r\   rX    s     L:Kf$:Ks   r   r   r   )rR  rS  )rR  )rP  rR  rS  )allr   zipis_floating_pointrE   float4_e2m1fn_x2randintuint8viewitemsizer8  float16tozeros
as_strided)rP  rQ  rR  rS  rT  needed_sizerl   s          r+   rand_stridedrh    s     K

$
L#d:KLLqP	
 E***]]3ekk&d5))*  ^^q  [[EMM&QTT U F [[&IF;-uLF&11r-   _Tc                  ,    [         R                  (       + $ r%   )r   r6  r}   r-   r+   check_dynamic_shape_capturerk    s    ....r-   patchesc                    ^ ^ [         R                  " T 5      S[        R                  S[        R                  S[
        4U U4S jj5       nU$ )Nr=   r>   r#   c            	         > [         R                  " 5        nT H,  u  p4nUR                  [        R                  " X4U5      5        M.     T" U 0 UD6sS S S 5        $ ! , (       d  f       g = fr%   )
contextlib	ExitStackenter_contextr
   r  )r=   r>   stackmoduleattrvalrH   rl  s         r+   _fn"_make_fn_with_patches.<locals>._fn  sT    !!#u%,!c##ELLs$CD &- t&v&	 $##s   :A
A))	functoolswrapsr!   r=   r>   ri  )rH   rl  rv  s   `` r+   _make_fn_with_patchesrz    sB    __R'277 'bii 'B ' ' Jr-   c                     U $ r%   r}   r*   s    r+   <lambda>r|    s    ar-   )
xfail_prop	decoratorcls
cls_prefix	fn_suffixr}  r~  c          	         [        U U R                   3U R                  0 5      nUR                  Ul        [	        U 5       H  nUR                  S5      (       a  [        X5      n[        U5      (       d  [        Xg[        X5      5        MK  U U 3n	[        U/UQ76 n
Xl        Ub&  [        X5      (       a  [        R                  " U
5      n
[        XiU" U
5      5        M  [        Xg5      (       a  M  [        Xg[        X5      5        M     U$ )Ntest_)rz   r   	__bases__r   dir
startswithgetattrcallablesetattrrz  hasattrunittestexpectedFailure)r  r  r  r}  r~  rl  DummyTestClassr.   rH   new_namenew_fns              r+   make_test_cls_with_patchesr    s     ZL7KN"0"9"9NC??7###BB<<gc.@A	{+H*288F&O%'"*A*A!11&9Ni.?@..N'#*<=   r-   c                 Z    [         R                  S:  a  U $ [        R                  " U 5      $ )N)r       sysversion_infor  r   rH   s    r+   skipIfNotPy311r    s$    
7"	==r-   c                 f    [         R                  S:  a  U $ [        R                  " S5      " U 5      $ )Nr       zRequires Python 3.12+r  r  s    r+   skipIfNotPy312r    s*    
7"	==01"55r-   c                     [         R                  S:  d  [         R                  S:  a  [        R                  " S5      " U 5      $ U $ )N)r       r  zRequires Python 3.12r  r  s    r+   skipIfOnlyNotPy312r    s7    
7"c&6&6&@}}34R88Ir-   c                 Z    [         R                  S:  a  [        R                  " U 5      $ U $ )Nr  )r  r  r  r  r  s    r+   xfailIfPy312r    s&    
7"''++Ir-   c                 f    [         R                  S:  a  [        R                  " S5      " U 5      $ U $ )Nr  zNot supported in Python 3.12+r  r  s    r+   skipIfPy312r  $  s*    
7"}}<=bAAIr-   c                     SU l         U $ NT)_expected_failure_dynamicr  s    r+   expectedFailureDynamicr  ,  s    #'B Ir-   c                     SU l         U $ r  )!_expected_failure_codegen_dynamicr  s    r+   expectedFailureCodegenDynamicr  2      +/B(Ir-   c                     SU l         U $ r  )!_expected_failure_dynamic_wrapperr  s    r+   expectedFailureDynamicWrapperr  8  r  r-   use_xlac                 $   [         R                  " S5        [        R                  " S5        [        (       a  [        R                  R                  S5        U (       a3  SS KJs  Jn  UR                  S[        UR                  5       5      5        g g )Ni9  r   )rE   manual_seedrandomseedr   torch_xla.core.xla_modelcore	xla_modelset_rng_stater   
xla_device)r  xms     r+   reset_rng_stater  =  sZ    	d
KK	r
		t--
s2==?34 r-   fr=   r>   c                     U " U0 UD6$ r%   r}   )r  r=   r>   s      r+   &_skipped_function_for_test_reconstructr  H  s     dfr-   c               #   4   #    / q  [         v   Sq g! Sq f = f7f)a@  
Context manager to capture graph modules compiled by invoke_subgraph_inductor_compile.

Usage:
    with _testing_capture_invoke_subgraph_inductor_compile_gms() as captured_gms:
        # code that triggers invoke_subgraph_inductor_compile
        pass
    # captured_gms will contain the list of captured graph modules
N)6_testing_invoke_subgraph_inductor_compile_captured_gmsr}   r-   r+   5_testing_capture_invoke_subgraph_inductor_compile_gmsr  Q  s&      >@:FDDAE>>s    )r0   )NNr   )Tg?)F)x__doc__ro  r   rx  loggingos.pathr   r  r2   r  typesr  collections.abcr   r   r   typingr   r   r   typing_extensionsr	   unittest.mockr
   rE   r    torch._dynamo.backends.debuggingr   torch._dynamo.output_graphr   r0   r   r   r   r   bytecode_transformationr   r   r   r   guardsr   r   r   r   r   r   r   r   r   r   
ModuleType__annotations__numpyModuleNotFoundErrorunsupportedthree	getLoggerr   logr!   rX   r,   r   r4   rI   rP   nnModulerd   rn   boolrr   r~   rc   r   floatr   CodeTyper   rx   r   r   r   r   rK   r
  r  r"  r'  r,  r/  rE  r   rG  rN  float32rR  rS  rh  ri  rk  rz  rz   r  r  r  r  r  r  r  r  r  r  r=   r>   r  r  contextmanagerr  r}   r-   r+   <module>r     s    
     	 
   9 9 ) ) '    6 2 8 8  A @ I I * #Et "
 $$	!t_>t# >t(; >. . .$"H%88??%(+%36%HK%	#Y%P	E3 	E4 	E 
 Au|| A A 
 A 
	cU38_	$tCH~	5
 

Es Eu||e'; E,3 
S 
 
s 
D 
 QR%
%
(+%
47%
<?%
JM%
%
P , < 
 
>A A(. . .C C Cc c , , ,    $'+ !8
8c8 8 *	8
 *8 8 
8B
(,U\\(:c3h !	
,
,
, 
, 	
,
 	
,  !&2
3-2SM2 ;;2 %,,	2
 2 \\2B T]/T /
	hr2v. 	# 	(2r6BR 	" "DO	  	
 d
 #s(+,hsCx.@@A 
>xB' HRV,< 6xB' 6HRV,< 68BF+ R0@ Xb"f% (2r6*: HRV$ "b&)9 xB/ HRV4D hr2v&6 8BF;K hr2v&6 8BF;K 
5T 5d 5B "3599 :> 6 Fy		@ F F  	Bs   +P P! P!