7h$SSKJr SSKrSSKrSSKJr SSKJrJrJ r SSK J r SSK J r SSKJr SSKJr SS KJrJrJrJr SS KJr SS KJr "S S \5r\"5r\R=\5S5r\R=\5S5r \RBS5r"\R=\RF5"\ "\SS95 \R=\RH5S5r%Sr&\S5r'Sr(Sr)Sr*g))contextmanagerN) DispatchKey)_ConstantFunction flat_apply to_graphable strict_mode)autograd_not_implemented)HigherOrderOperator)FakeTensorMode)get_proxy_slotPreDispatchTorchFunctionModeProxyTorchDispatchModetrack_tensor_tree)_pytree)"is_traceable_wrapper_subclass_typec4^\rSrSrU4SjrU4SjrSrU=r$)ExportTracepointc$>[TU]S5 g)N_export_tracepoint)super__init__)self __class__s P/var/www/fran/franai/venv/lib/python3.13/site-packages/torch/_export/wrappers.pyrExportTracepoint.__init__s -.c$>[TU]"U0UD6$N)r__call__)rargskwargsrs rr!ExportTracepoint.__call__sw000r)__name__ __module__ __qualname____firstlineno__rr!__static_attributes__ __classcell__)rs@rrrs/11rrc[R"URRX45up4URR S[ X45n[ XSURS9$)N call_functionconstanttracer)pytreetree_mapr0 unwrap_proxy create_proxyrr)moder"r#p_argsp_kwargsproxys rexport_tracepoint_dispatch_moder9%sRt{{'?'?$PF KK $ $+V E T4 LLrc@U UsSSS5 $!,(df  g=fr r%)r5r"r#s r"export_tracepoint_fake_tensor_moder;.s  s cURU5nURU5nUR5 [U0UD6 UsSSS5 $!,(df  g=fr )unwrap_tensorsredispatch_to_nextr)ctxr"r#unwrapped_argsunwrapped_kwargss rexport_tracepoint_functionalrB4sL''-N))&1   !N?.>? " ! !s A  AT)deferred_errorcU$r r%)r"r#s rexport_tracepoint_cpurECs Krc:^^^^ [U[RR5(deTS:wde[RR R UT5nU4Sjm SmUU4SjnUUU 4SjnURUSS9nURUSS9nXg4$)NcX>UT;aTUSU:XdeTUSU:XdeXS.TU'g)Nin_specout_spec)rIrJr%)pathrIrJmodule_call_specss rupdate_module_call_signatures6_wrap_submodule..update_module_call_signaturesMsJ $ $$T*95@ @@$T*:6(B BB.5"L$rc UHJn[U[R[[[ [ 45(aM9UcM>[SU35e g)NzGOnly Tensors or scalars are supported as pytree flattened inputs, got: ) isinstancetorchTensorstrintfloatboolAssertionError) flat_argsas rcheck_flattened(_wrap_submodule..check_flattenedSsEAq5<<c5$"GHHAI$]^_]`arc>[R"X45up4T"U5 [USTS.6n[R"X45upX4$)Nmodule_call_inputskindrKr1 tree_flattenrtree_unflatten)moduler"r#rXrIrZrKs rpre_hook!_wrap_submodule..pre_hookZsJ#00$@  "& 8LSWX ,,Y@ |rc>[R"X45upE[R"U5upgT"U5 [UST S.6nT "T XW5 [R"Xg5$)Nmodule_call_outputsr^r`) rcr"r#res_rIflat_resrJrZrKrMs r post_hook"_wrap_submodule..post_hookas](($8 #005!%x6KRVW%dG>$$X88rT) with_kwargs) rPrQnnModulefx graph_module _get_attrregister_forward_pre_hookregister_forward_hook) modrKrL submodulerdrk pre_handle post_handlerZrMs `` @@r_wrap_submoduleryHs c588?? + ++ + 2::%%//T:IM 944X44PJ11)1NK  ""rc## /nUHnUR[XU55 M Sv UHnUR5 M g!UHnUR5 M f=f7fr )extendryremove)fpreserve_signaturemodule_call_signatureshandlesrKhandles r_wrap_submodulesrnsVG&D NN?14JK L' F MMOgF MMOsA&(AA&A##A&cSnXlU$)Nc[X5$r r)rr"s rcall'_mark_strict_experimental..call|s 4&&r)r!)clsrs r_mark_strict_experimentalr{s'L JrcUS-n[URU5(a0[URU5U:XdeURSUS05$UR U5n[ URXB5 URSUS05$)a This is a wrapper utility method on top of tracer to cache the already registered subclass spec attribute. This is useful because Subclass.__init__ will be same for each subclass. By default, fx will create multiple attributes/proxies for given attribute. 0get_attrr%)hasattrrootgetattrr4get_fresh_qualnamesetattr)r0namespecfx_namequalnames r'_register_subclass_spec_proxy_in_tracerrsSjGv{{G$$v{{G,444"":wB??((.H FKK(   z8R <._is_inits|9 z 99rztorch._export.wrappers.mark_constructor_exportable_experimental can only be applied on subclass tensor.__init__But, you are adding it on z which is not supported. If __init__ doesn't exist on your subclass, please add it. Look at DTensor.__init__ implementation for examplec>^[[US55(dJTRRS5(deTRS[ S5*n[ SUS35eT"U0UD6 [ RR5(dg[ RR5nUVs/sHn[U[5(dMUPM nn[ U5S::dS[ U535e[ U5S:XagUSnURmUSn[[USS5U45upxSR!TRR#5R%S55n TR'U 5n [)TR*X5 TR-S U S 05n [.R0"[ R2U4S jU5n [ R4R6R9[;[U555up[U5R<R#5S -n[?TX5nTR-S [@UU /U Q705n[CUUSTS9 gs snf)Nrrz5Applying mark_constructor_exportable_experimental on z is not valid as it is not a traceable tensor subclass. Please look at DTensor.__init__ implementation as an example of proper usage of this API.z6Expected only one PreDispatchTorchFunctionMode, found ri.rr%c0>[UT5R$r )r r8)xr0s rTmark_subclass_constructor_exportable_experimental..wrapper..sN1f$=$C$Cr_const_func_specr-r.)"rtyper(endswithlen RuntimeErrorrQ_C_is_torch_function_mode_enabled overrides _get_current_function_mode_stackrPrr0rtuplejoinlowersplitrrrr4r1 tree_map_onlyrRutilsrrarr&rrr)r"r#obj_nametorch_function_mode_stackr5pre_dispatch_tf_modessubclassrXrIconstructor_spec_namer spec_proxyflat_proxy_argsri func_spec!fxable_constructor_call_spec_namefunc_spec_proxy inner_proxyr0constructor_subclasss @rwrapperBmark_subclass_constructor_exportable_experimental..wrappersn1$tAw-@@'44==jII II+889KC O;KLHGzR}~  d-f-xx7799 $)OO$T$T$V!2! 1$ <= 1 ! % &! + a CCH]D^C_ ` a + $ % * $Q'7)5ab?F*CD # - - 3 3 5 ; ;C @! ,,-BC X/((Xr2F  .. LLCY {{**77 d8n -  N # # ) ) +.@ @ *B 5 ))   j ;? ;   (K$vNq! s 7I5I5)rr&)rrrs` r1mark_subclass_constructor_exportable_experimentalrsO,: ( ) )))=)F)F(GH} ~  EN Nr)+ contextlibrrQtorch._custom_opstorch._Cr"torch._higher_order_ops.flat_applyrrr#torch._higher_order_ops.strict_moder torch._higher_order_ops.utilsr torch._opsr torch._subclasses.fake_tensorr "torch.fx.experimental.proxy_tensorr rrr torch.utilsrr1torch.utils._python_dispatchrrrpy_implr9r;py_functionalize_implrBAutogradCPUrEryrrrrr%rrrs!%   <B*8 *K1*1&'23M4MN+, ))*;//0/E KOO,-##L  ="gr