|YhFddlmZddlZddlmZddlmZmZmZm Z ddl m Z erddl m Z ddlZddlZddlZddlZddlZddlZddlZddlZddlmZe dZe d Zejdd ZejddZGddejjZdS)) annotationsN) nullcontext)AnyCallable TYPE_CHECKINGTypeVar) ParamSpec)Sequence)torch_function_passthrough_P_Rreturndict[Any, Any]c  ttjftjtjjftjjtjjjftjtjjftjtjjftjtjjfg}itjjtjj tjj tjj tjj tjj tjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjjtjj tjj tjj!tjj!tjj"tjj"tjj#tjj#tjj$tjj$tjj%tjj%tjj&tj'j(tjj)tj'j)i}|D]D\}}|j*D]7}|j+,|||j+,|<8Et[tjD]T}|tjj*vr?tjj+,||t]tj|<Utjj/j*D]^}t]tj|dpt]t|}tjj/j+,|||<_|S)z} Mapping of torch API functions to torch._refs functions. E.g. torch_to_refs_map()[torch.add] == torch._refs.add N)0torch_refsnn functionalspecialfftlinalgTensor __invert__ bitwise_not__xor__ bitwise_xor__and__ bitwise_and__or__ bitwise_or__eq__eq__rsub__rsub __rtruediv__rtruediv __floordiv__ floor_divide __rfloordiv__ rfloordiv__pow__pow__rpow__rpow new_emptynew_full new_zerosnew_onesfill_zero_to sum_to_sizecopy__primscopy_toresize__all____dict__getdirgetattr _conversions)modulesr mod_torchmod_refss tensor_attrs f/var/www/tools.fuzzalab.pt/emblema-extractor/venv/lib/python3.11/site-packages/torch/_prims/context.pytorch_to_refs_maprHs;   5;>"  ekn78  +, EKO$ u{)* G !8 ek5  ek5  U[3   U[^   u{/   !5;#7  !5;#;  "EK$9  eko  u{/   5  u{3   5  u{3  EK-!"  EK-#$    %+"9 EL0 U\0-A0 'DD 8! D DA+3+<+@+@+C+CAi $$Q'' ( ( D  FF  # # #*/+*>*B*B1*E*EAgelA&& '[ % -BBelAt44Iq8I8I 1:>>qAA+ Hset[Any]c8dtjjDS)zJ Set of all prim functions, e.g., torch._prims.add in all_prims() cVh|]&}tjj|'S)rr8r<r=).0rEs rG zall_prims..Ws+ G G GQEL ! % %a ( ( G G GrI)rr8r;rMrIrG all_primsrPRs H G%,2F G G GGrIc2eZdZdZddefdd Z dddZdS) TorchRefsModeaB Switches the interpretation of torch.* functions and Tensor methods to use PrimTorch refs in torch._refs. (Direct calls to _refs are unaffected.) >>> # xdoctest: +SKIP >>> with TorchRefsMode(): ... torch.add(x, y) # calls torch._refs.add(x, y) By default, this context manager will fall back on the torch.* if the ref does not exist; set strict=True to error if this occurs. If the ref exists we still would like to fall back on the torch.* sometimes, this behavior can be customized by passing a function to should_fallback_fn. FcdS)NFrM)_s rGzTorchRefsMode.lsUrIstrictboolshould_fallback_fnCallable[..., bool]prims_mode_clstyperNonec0||_||_||_dS)N)rVrXrZ)selfrVrXrZs rG__init__zTorchRefsMode.__init__is!  "4,rIrMN orig_funcCallable[_P, _R]typesSequence[type]args Sequence[Any]kwargsdict[str, Any] | Nonerc||i}|tvs|tvr4|5||i|cdddS#1swxYwYt}||d}|Et |t jjr&t j j |d}n~||t |t jj r]t|dd}|#|j rt||j dd}|%t j j |d}|C||||||r||i|S|5||i|cdddS#1swxYwY|jr/t!dt j|||i|S)Ndefaultrzno _refs support for )r rPrZrHr= isinstancer_ops OpOverload_decompdecomposition_tableOpOverloadPacketr?_dirrXrV RuntimeError overrides resolve_name)r^r`rbrdrfmappingfuncris rG__torch_function__z TorchRefsMode.__torch_function__ssa >F 2 2 2i9;;6N6N$$&& 2 2 y$1&11 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2 2#%%{{9d++ !)Y^A->EE"}8<rs""""""""""""888888888888'''''')((((((  ::::::Yt__ WT]] 3 3 3 3 l HHHHD*D*D*D*D*EO5D*D*D*D*D*rI