EiBSSKrSSKrSSKJrJrJrJr SSKrSSKJ r SSK J r \R"\ 5r"SS5r"SS5r"S S \5r"S S 5r\"5rS rS\R*S\4SjrS\S\\\R*44SjrS S\\4SjjrSrS\4SjrS\\4SjrS\S\4SjrS\S\4SjrS\S\ \\44Sjr!S\R*S\4Sjr"Sr#S\4Sjr$g)!N)AnyOptionalProtocolUnionparse_namespace_disable_current_modesc^^\rSrSrS\S\S\\R4Sjr U4Sjr Sr Sr U=r $) FakeScriptObject wrapped_objscript_class_namexc[RUSU5 [RUSU5 [5 [R"U5nSSS5 [RUSW5 g!,(df  N&=f![ [ 4a,n[RSU[U55 UnSnAN`SnAff=f)NrrzUnable to deepcopy the custom object %s due to %s. Defaulting to the user given object. This might be dangerous as side effects may be directly applied to the object.real_obj) object __setattr__r copydeepcopy RuntimeError TypeErrorlogwarningstr)selfrrrres _/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/_library/fake_class_registry.py__init__FakeScriptObject.__init__s 4 <4!46GH ')==+* 4X6*)i(  KK!"A  H s4 BA1B1 A?;B?BB>"B99B>ch>[TU]U5$![an[SUS35UeSnAff=f)Nz%Tried to call __getattr__ with attr 'aE' on a FakeScriptObject, implying that you are calling this inside of a fake kernel. The fake kernel should not depend on the contents of the OpaqueObject at all, so we're erroring out. If this attr is a method or constant attribute, you can allow this member access by registering it via `register_opaque_type(members=...)`.)super__getattribute__AttributeError)rnamer __class__s rr#!FakeScriptObject.__getattribute__$sM 7+D1 1  7v>JJ  s 1,1c [SUS35e)Nz%Tried to call __setattr__ with attr 'a3' on a FakeScriptObject, implying that you are calling this inside of a fake kernel. The fake kernel should not depend on the contents of the OpaqueObject at all, so we're erroring out. If you need thisfunctionality, consider creating a custom TorchBind Object instead(but note that this is more difficult).)r$)rr%values rrFakeScriptObject.__setattr__1s 3D6:6 6  )__name__ __module__ __qualname____firstlineno__rrrtorch ScriptObjectrr#r__static_attributes__ __classcell__)r&s@rr r s:77367;CEDVDV;W7*   r+r cJ\rSrSrS\S\S\\R4Sjr Sr Sr g) FakeScriptMethod< self_fake_obj method_nameschemac(XlX lX0lgN)r8r9r:)rr8r9r:s rrFakeScriptMethod.__init__=s +& r+cPSSKJn U"URUR/UQ70UD6$)Nr)call_torchbind)!torch._higher_order_ops.torchbindr?r8r9)rargskwargsr?s r__call__FakeScriptMethod.__call__Gs(Dd00$2B2BTTTVTTr+)r9r:r8N) r-r.r/r0r rrr1FunctionSchemarrCr3r,r+rr6r6<s4'--. Ur+r6c@\rSrSr\S\R 4Sj5rSrg)HasStaticMethodFromRealMrcgr<r,)clsrs r from_real!HasStaticMethodFromReal.from_realNs r+r,N) r-r.r/r0 classmethodr1r2rKr3r,r+rrGrGMs! !3!3  r+rGc\rSrSrS SjrS\S\4SjrS\S\4Sjr SS\SS4Sjjr S\S\4S jr S S jr S\SS4S jr S rg)FakeClassRegistrySreturnNc0Ulgr<_registered_classrs rrFakeClassRegistry.__init__Ts 13r+ full_qualnamecXR;$r<rSrrWs rhas_implFakeClassRegistry.has_implWs 6 666r+cBURU5 URU$r<)_check_registeredrTrYs rget_implFakeClassRegistry.get_implZs! }-%%m44r+czURU5(a[RSUU5 X RU'g)NzE%s is already registered. Previous fake class is overridden with %s.)rZrrrT)rrW fake_classs rregisterFakeClassRegistry.register^s6 == ' ' KKW  1;}-r+cURU5(d[RSU5 gURR U5$)NziCannot deregister %s. Please use register_fake_class to register it first. Or do you dereigster it twice?)rZrrrTpoprYs r deregisterFakeClassRegistry.deregistergs>}}]++ KK2  ))--m< XR;a[US35eg)NzH is not registered. Please use register_fake_class to register it first.)rTrrYs rr]#FakeClassRegistry._check_registeredts+  6 6 6 /!ij  7r+rS)rQNr<)r-r.r/r0rrboolrZrr^rbrfrir]r3r,r+rrOrOSsk47c7d75c5c5;c;t;==='str+rOc[U[5(d [S5eUHUn[U[5(d [S5e[U5S:Xa[US[5(aML[S5e g)NzExpect flat x to be a tuple.z&Expect flat x to be a tuple of tuples.rzXExpect element of flat x to be a tuple of two elements with first element being a string) isinstancetuplerlenr)flat_xtps r_check_valid_flat_script_objrusi fe $ $9::"e$$GH H2w!|:beS#9#9j  r+rrQc[US5(dgUR5S;a[SUR535eUR5S:H$)N tracing_modeF)realfakez0tracing_mode can be either real or fake but got rx)hasattrrwAssertionError)rs rtracing_with_realr|sU 1n % %~~//>q~~?O>P Q   >> v %%r+c SSKJs Jn SSKJn [ U5(aU$SSKJnJnJ nJ nJ n SSK J n [U5n U"U 5(aUcUOU"U 5n [U"5X5n U"U 5n U c[!SU S35eU R"HTnU "5 [%X5(d['SU SUS 35e[(R+X[-X55 SSS5 MV U $U"5 UR/5nSSS5 [1W5 U SS KJn [7U5VVs0sH0unn[9U[:R<5(dM'U"U5U_M2 nnn[7U5VVs0sHDunn[9U[:R<5(dM'UU"U5U:wdM8UUU"U5_MF nnn[?U5S:a[@RCS UUU5 URE[:R<S U5nSSS5 [GU5RIW5n[UURK5RM5U5n URO5Hn[-UUS5nUb[QU5(d[SS USUS35e[-UU5nSn[9U[:RT5(a URVn[(R+U U[YU UU55 M1SknUU;dM[@RCSUU5 M U $!,(df  GM=f!,(df  GN9=fs snnfs snnf!,(df  GNN=f)Nrr )FakeOpaqueObjectget_opaque_obj_infoget_opaque_type_nameis_opaque_type OpaqueTypeStr)unset_fake_temporarilyzopaque_info for type z must not be NonezOpaque object of type 'z ' was specified to have member 'z1', but this doesn't actually exist in the object.)_tensor_storagezDetected script object %s has aliasing relationship among its tensors. Flattened obj: %s. Aliasing tensor indices: %s. This is not supported and may cause unexpected behavior.c [R"UR5UR5URUR UR URS9$)N)devicedtype requires_gradlayout)r1 empty_stridedsizestriderrrr)ts r#maybe_to_fake_obj..s=%--FFHHHJ88''"#//88 r+zExpect z to be a callable but got .> __getstate__ __setstate____obj_flatten__z.fake object of %s doesn't implement method %s.)-torch.utils._pytreeutils_pytreetorch.utils._python_dispatchr r|torch._library.opaque_objectr~rrrrtorch._subclasses.fake_tensorrtyper r{membersrzrrrgetattrrrutorch._higher_order_ops.utilsr enumeraterpr1Tensorrrrr tree_map_only"_find_fake_class_for_script_object__obj_unflatten___typequalified_name _method_namescallabler ScriptMethodr:r6) fake_moderpytreer r~rrrrrx_type type_namefake_x_wrapped opaque_info attr_namersriinp storage_map alias_mapfake_flattenedfake_xr%attr real_attr method_schemaoverride_skip_lists rmaybe_to_fake_objrsD)(CE !WFf%&YM4H4P )*:*ga>ST *)- $ %&&(F& %V,  E(//FAsc5<<0($a'/ (//FAsc5<<05  45:5;s344/  9~! O $11  N7P4A6HHX%faggi.F.F.H!LN!vtT*  D>>"WTF2LTFRS#TUU4(I=AM)U%7%788 ) 0 0     }E  "U -- LaQUV-". m*)& % YsVAL#3L6M.&M M%M6&M M1MA M# L3 6 M M M#rac:^S[4U4SjjnUcU$U"U5$)a Register a fake implementation for this class. It's in the same spirit of registering a fake implementation for an operator but with the difference that it associates a fake class with the original torch bind class (registered with torch::class_). In this way, torch.compile can handle them properly in components such as Dynamo and AOTAutograd. This API may be used as a decorator (see example). For the fake class, users are required to provide a from_real classmethod that takes a real object and returns an instance of the fake class. All tensors in the fake object should also be properly fakified with to_fake_tensor() in from_real. Examples: # For a custom class Foo defined in test_custom_class_registration.cpp: TORCH_LIBRARY(_TorchScriptTesting, m) { m.class_("_TensorQueue") .def(torch::init()) .def("push", &TensorQueue::push) .def("pop", &TensorQueue::pop) .def("top", &TensorQueue::top) .def("size", &TensorQueue::size) .def("clone_queue", &TensorQueue::clone_queue) .def("__obj_flatten__", &TensorQueue::__obj_flatten__) .def_pickle( // __getstate__ [](const c10::intrusive_ptr& self) -> c10::Dict { return self->serialize(); }, // __setstate__ [](c10::Dict data) -> c10::intrusive_ptr { return c10::make_intrusive(std::move(data)); }); }; # We could register a fake class FakeTensorQueue in Python as follows: import torch @torch._library.register_fake_class("_TorchScriptTesting::_TensorQueue") class FakeTensorQueue: def __init__(self, queue): self.queue = queue @classmethod def __obj_unflatten__(cls, flattened_ctx): return cls(**dict(ctx)) def push(self, x): self.queue.append(x) def pop(self): return self.queue.pop(0) def size(self): return len(self.queue) In this example, the original TensorQeue need to add a __obj_flatten__ method to the class TensorQueue and the flattened result is passed into FakeTensorQueue's __obj_unflatten__ as inputs to create a fake class. This protocol allows pytorch to look at the contents of the script object and properly handle them in the subsystems like dynamo, aot_aotugrad or more. rach>[T5up[RRX5 [ U[ S5nU(d[ US[ S35e[UR[ [5(d[ [ S35e[R[T5U5 U$)Nz doesn't define a classmethod rz method is not a classmethod.) rr1_C _get_custom_class_python_wrapperr_CONVERT_FROM_REAL_NAMErrp__dict__rMglobal_fake_class_registryrb_full_qual_class_name)ransr% from_methodqualnames rinner"register_fake_class..innerSs"8, 11";j*A4H ,<=T[R[U55$r<)rrfr)rs rderegister_fake_classrls % 0 01Fx1P QQr+c,[RU5$r<)rrZrWs rhas_fake_classrps % . .} ==r+cN[U5(dg[RU5$r<)rrr^rs rfind_fake_classrts! - ( ( % . .} ==r+rc2[U5upSU-S-U-$)Nz__torch__.torch.classes.rr)rrr%s rrrzs#x(HB % *S 04 77r+objc[U[R5=(a- UR5R 5R S5$)Nz__torch__.torch.classes)rpr1r2rr startswith)rs r_is_script_objectrs?  U     ))+ $ $ & 1 1! r+rWctURS5n[U5S:wa[SU<S35eUup#pEnXV4$)NrzCould not split full_qualname=z, expected 5 parts)splitrrr{)rWsplits_torch _torch_ns_classesr class_names r_ns_and_class_namersI   %F 6{a> /??QRSS28/FxZ >r+c UR5R5n[U5up#[U5nUc[ SUSUSUS[ S3 5eU$)Nz ScriptObject's zA haven't registered a fake class. Please use register_fake_class(z::a) to annotate a fake class for the script obj. Specifically, create a python class that implements a fake version for all the methods that're used in the program and put annotated class in the program e.g. after loading the library. The fake methods can be written in the same way as a meta kernel for an operator but need to additionally simulate the object's states. Be sure to add a z; classmethod to enable creating a fake obj from a real one.)rrrrrr)rrWrrras rrrsyGGI,,.M' 6NB /J}o.//1d"ZLA?@W>WX>  ?   r+rcz^[U5n[U[S5nU(d[US[S35e[R R RUS5m[R R RU4Sj5 URU5sSSS5 $!,(df  g=f)Nz must define a classmethod z2 that converts the real object to the fake object.c>T$r<r,)ctxsrr%_fake_obj_from_real..sr+) rrrrr1_library fake_impl FakeImplCtxset_ctx_getterrK)rrrafrom_real_methodrs @r_fake_obj_from_realrs3A6Jz+BDI l56M5N@ B  .. " " . .y$ ?C  ! ! 0 0 =##A& > = =s B,, B:r<)%rloggingtypingrrrrr1torch._library.utilsrrr getLoggerr-rr r6rGrOrrur2rmr|rrrrrrrrrqrrrrr,r+rrsT 11 0?!+ + \UU" h %%P/0 &++&&u u U// /0upYh7N.OYxR>T>>hsm> 8C8C8 34ceCHo%*<*<". ' 'r+