+ j+f^RIHt^RIt^RIt^RIHtHt^RIt^RIH uH t ^RI H t HtHtHtHtHtHt^RIHt]'d^RIHtHtHt^RIHt^RIHt]P:P<P>tR R lt R R lt!R Rlt"RRlt#RRlt$RRlt%RRRllt&!RR]4t'R#)) annotationsN)Any TYPE_CHECKING) FakeTensorFakeTensorModeis_fake_tensormaybe_get_fake_modeMetadataMismatchErrortree_flatten_onlyUnsupportedFakeTensorException)TorchDispatchMode)CallableMappingSequence) OpOverload)PyTreec$V^8dQhRRRRRR/#outputsrinputsreturnbool)formats"]/home/luca/screen-checking/.venv/lib/python3.14/site-packages/torch/_subclasses/fake_utils.py __annotate__rs!  & & T ca\\PV4Uu0uFDp\PP V4'gK*VP 4P kKF upo\;QJd7V3Rl\\PV44F 'gK R# R#!V3Rl\\PV444#uupi)c3<"TFIp\PPV4;'dVP4PS9xKK R#5iN)torch_C _has_storage_typed_storage_cdata).0outinput_storagess& r 'outputs_alias_inputs..%sF;C c"TTs'9'9';'B'Bn'TT;s *A&ATF)r r"Tensorr#r$r%r&any)rrinpr)s&& @routputs_alias_inputsr/s%U\\6::C 88  % $##:N 3$U\\7;3333$U\\7;  s 'C Cc$V^8dQhRRRRRR/#rr)rs"rrr+s&YYYY4YrcHa\\PV4Uu0uFp\V4kK upo\;QJd7V3Rl\\PV44F 'gK R# R#!V3Rl\\PV444#uupi)c3@<"TFp\V4S9xK R#5ir!)id)r'r( input_idss& rr*%outputs_are_inputs..-sX/Wr#w)#/WsTF)r r"r,r3r-)rrr.r4s&& @routputs_are_inputsr6+so$5ellF$KL$KSC$KLI 3X/@w/WX33X3X3X/@w/WX XXMsBc V^8dQhRRRR/#)rrrrrr)rs"rrr0s  V  rc\4p\\PV4F^p\PP V4'gK*VP 4PpW19dR#VPV4K` R#)TF) setr r"r,r#r$r%r&add)rstoragesr(stors& routput_alias_each_otherr=0sduH w7xx$$S)) !!#**   T 8 rc 0V^8dQhRRRRRRRRRRRR /#) rcontextstrreal_outrreal_infake_outfake_inrNoner)rs"rrr<sD           rc \W4p\W44pWV8wd\V RV RV 24h\W4p\W44pWx8wd\V RV RV 24h\V4p \V4p W8wd\V RV RV 24hR#)z( mismatch in outputs_alias_inputs check  != z& mismatch in outputs_are_inputs check z, mismatch in outputs_alias_each_other check N)r/r r6r=) r?rArBrCrD r_aliasing f_aliasing r_identity_eq f_identity_eqr_output_alias_each_otherf_output_alias_each_others &&&&& r_check_alias_inforN<s&h8J%h8J#i? |4PZ| \  'x9M&x9M%#i=m_DQ^P_ `  !8 A 7 A =#iC().G-H J  >rc(V^8dQhRRRRRRRR/#) rfuncridxinte Exceptionrrr)rs"rrrZs( trcV\PPJgV\PPJdVR9dR\ V49dR#V\P PJgV\P PJdVR9dR\ V49dR#V\PPJdVR9dR\ V49dR#R#)DevicesTF)rV)r)aten#_scaled_dot_product_flash_attentiondefault_flash_attention_forwardrepr'_scaled_dot_product_efficient_attention_efficient_attention_forward#_scaled_dot_product_cudnn_attention)rPrQrSs&&&r is_sdpa_errorrbZs D<<DD Dt44<<< 6M a  D@@HH Ht88@@@ 6M a  88@@@ 6M a  rc V^8dQhRRRR/#)rten_listzlist[FakeTensor | Any]rz%list[FakeTensor | torch.Tensor | Any]r)rs"rrrvs DD$D*Drc~a\RV4R4pVfV#\V4pVfV#VPPpVPpVP P 4UUu/uFwrVWebK pppVPP 4UUu/uFwrVWebK ppp.p VEFvp \V 4'd V P\P8wdV PV 4KGVPV P44p V fRMVPV 4p V fV PV 4KRoRV3Rllp V !V P44pV P Uu.uF q!V4NK ppV P#4Uu.uF q!V4NK ppS'dV PV 4EK\P$!.V P&V P(R7pVP+V VVVR7V PVP-44EKy V #uuppiuuppiuupiuupi)aU Attempt to convert fake tensors to a corresponding real tensor with the correct underlying storage by looking up the FakeTensorMode meta to real storage mapping. On failure to find the storage mapping, the FakeTensor will remain in the list. Note: this is not currently optimized (makes copies of the meta converter internal dictionaries) c3L"TFp\V4'gKVxK R#5ir!)r)r'items& rr*+try_convert_fake_to_real..sJ^D5Is$ $NFc V^8dQhRRRR/#)rsztorch.SymInt | intrrRr)rs"rr.try_convert_fake_to_real..__annotate__s  ,  rct<\V\P4'gV#VPpVfRo^#V#)NT) isinstancer"SymInthint)rjrocontains_unhinteds& r map_symint,try_convert_fake_to_real..map_symints5a..66D|$(!Kr)dtypedevice)storage_offsetsizestride)nextr fake_tensor_convertermeta_converter describer storage_memoitemslookup_storagerlayoutr"stridedappendgetuntyped_storagerushaperwemptyrsrtset_clone)rd fake_tensor fake_moderzdesckvstorage_to_keykey_to_real_storager(tkey real_storagerq stor_offsetrjrvrw new_tensorrps& @rtry_convert_fake_to_realrvsJJDQK#K0I44CCN  # #D'5'B'B'H'H'JK'Jtqad'JNK,0,?,?,E,E,GH,GDA14,GH C a  AHH $= JJqM   !2!2!45"{t0C0G0G0L   JJqM !  !!1!1!34 '(ww/w! 1 w/)*4A*Q-4  JJqM [[ ''88  &  :##%&UX J_LH404s+ H) H//H5H:c8V^8dQhRRRRRRRRR RR RR RR R /#)rrAz torch.TensorrCrr?r@sizesrstridesru requires_gradrrEr)rs"rrrsX       rcvV'd*VPVP8wd\V R24h\PP V4'd5VP 4pVP 4pWx8wd\V R24h\P PPVVVVRR7R#)z mismatched requires_grad-ness of outputs. This usually means that you have added autograd support for your operator at a dispatch key other than Autograd, which will lead to problemsz mismatched storage offsetT) check_sizes check_stridesallow_rhs_unbackedN) rr r"r#r$ru_primsutilscompare_tensor_meta) rArCr?rrrurr_offsetf_offsets &&&&&&& r_check_fake_real_tensorsrs  ! !X%;%; ;')./  xxX&&**,**,  '7)3M(NO O LL** +rcRa]tRt^tR RRRRRR/RV3RlllltR RRlltR tV;t#) CrossRefFakeModerTcheck_aliasingonly_check_ops_with_metac ,V^8dQhRRRRRRRRRR/#) r ignore_op_fnz#Callable[[OpOverload], bool] | NonerrrrrrEr)rs"rrCrossRefFakeMode.__annotate__sCAA9A A  A #' A Arc d<\SV`4VeTMRVnW nW0nW@nR#)NcR#)Fr)fns&r+CrossRefFakeMode.__init__..sUr)super__init__rrrr)selfrrrr __class__s&&$$$rrCrossRefFakeMode.__init__s7 (4L:J +,(@%rc ,V^8dQhRRRRRRRRR R /#) rrPrtypeszSequence[type]argszSequence[object]kwargszMapping[str, object] | Nonerobjectr)rs"rrrsC\\\\ \ , \  \rc  T;'g/pRpRp/pV\PP\PP\PP 39EdvVP V4'Eg^VP'd1\PPPV4'Ed\PPVP9d\PPVP9d\PP VP9d^RIHp\'V!4R7;_uu_4p \(P*!\P,\.P0!V P2RR7W434wrg\4P6!4;_uu_4V!V/VBpRRR4RRR4RV R2p V!V/VBp VEeD\(P:!V 4p \(P:!V4p \=V 4\=V 48wd&\?V R\=V 4 R\=V 4 24hVP@'d\CWW43WVV34\E\G\(P:!V 4\(P:!V444Fvwpwpp\IV\P,4pV\IV\P,48wd\?V R 24hV'gKZ\KVVRVPLRRR 7Kx V # +'giELq;i +'giEL};i \8dELi;i \NdKp\QYT4'dRp?K\=T 4^8XdT R T 2M T R T R T 2p\ST4ThRp?ii;i)N)ShapeEnv) shape_envT) static_shapeszWhen comparing the output of z* on FakeTensor and concrete Tensors, foundz mismatch in number of returns rGz$ mismatched number of tensor outputs)rrrurz mismatched tensor metadata: z' mismatched tensor metadata for output[z]: r)*rZ lift_freshr\lift_fresh_copyrsource_Storage_storage_offsetrrr" _subclasses fake_implshas_metaTagdynamic_output_shapetags inplace_viewdata_dependent_output%torch.fx.experimental.symbolic_shapesrrpytree tree_map_onlyr, functoolspartial from_tensorwarningscatch_warningsr tree_leaveslenAssertionErrorrrN enumerateziprmrrrTrbr )rrPrrrfake_r fake_args fake_kwargsrrr?rr_flatf_flatrQr_outf_outr_is_tenrS error_messages&&&&& r__torch_dispatch__#CrossRefFakeMode.__torch_dispatch__sE2&( ,. ''$$,, 77  %%d++111$$//88>> ..dii? &&dii7 //tyy@ G #hj99Y-3-A-A !)))*?*?tT.*I "0022!%y!@K!@3 :,D62   $ !& !  ''*F''/F6{c&k)$i>s6{m4PSTZP[}]"""!K8P(1F&&q)6+=+=f+EF(#^eU&eU\\:z%>>(")#GH8J0!!"&$($6$6+/*. (6e322 :992  N%J(A66$ #6{a/ 'i'DQCH$+9,STWSXX[\][^!_& 4MBIJsgMA'M L3 MMM-3 M > M M MM M*)M*- O8N=-N==O)rrrrr!)rN)__name__ __module__ __qualname____firstlineno__rr__static_attributes__ __classcell__)rs@rrrs<A# A $ A *. AA \\rr)TFTT)( __future__rrrtypingrrr"torch.utils._pytreer_pytreertorch._subclasses.fake_tensorrrrr r r r torch.utils._python_dispatchr collections.abcrrr torch._opsrr_opsopsrZr/r6r=rNrbrrrrrrrs"% $$;;;%* zz~~ Y  <8DNBm(mr