SjUddlZddlZddlmZddlmZmZmZddlm Z da ejdze d<ejddZGdd eZd Zd Zy) N)_len_torch_function_stack) _pop_mode _push_modeTorchFunctionModecontext_decoratorCURRENT_DEVICEchtjtjtjtjtj tj tjtjtjtjjtjjtjtjtjtj tj"tj$tj&j(tj*tj,tj.tj0tj2tj4tj6tj8tj:tj<tj>tj@tjBtjDtjFtjHtjJtjLSN)'torchemptyempty_permuted empty_stridedempty_quantizedonesarangebartlett_windowblackman_windoweyefftfftfreqrfftfreqfullhamming_window hann_window kaiser_windowlinspacelogspacenested nested_tensorrandrandnrandintrandpermrangesparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensor tril_indices triu_indiceszerosasarraytensor as_tensor scalar_tensorR/var/www/html/ia-enhanced/venv/lib/python3.12/site-packages/torch/utils/_device.py_device_constructorsr7 sF1  1  1   1   1   1   1  1  1   1   1   1   1  1  1  !1 " #1 $ %1 &  ""'1 ,  -1 .  /1 0  11 2 31 4  51 6 71 8 &&91 : ;1 < =1 > ?1 @ A1 B C1 D E1 F  G1 H  I1 L  M1 N O1 P Q1 1r5c(eZdZddZdZdZddZy) DeviceContextNcFtj||_d|_yr )r device prev_mode)selfr;s r6__init__zDeviceContext.__init__Esll6* /3r5ct|_|jatt Dcgc] }t }}t |t|D]%}t|tr||_ t |'ycc}wr ) r old_devicer;r&rrrreversed isinstancer9r<)r=_ cur_stackmodes r6 __enter__zDeviceContext.__enter__Ism( +00I0K*LMQY[M M4Y' !D$ .!%4  ! NsA?c|jag}ttdz D]8}t }t |t r td|j|:tdkDr%t }t |t s td|jt|jt|D] }t|y)Nr z@Found nested DeviceContext on the mode stack where none expectedrz8Expected a DeviceContext at the bottom of the mode stack) r@r r&rrrBr9AssertionErrorappendr<rrA)r=exc_typeexc_valexc_tbrDrCrEs r6__exit__zDeviceContext.__exit__[s 02Q67 #A;D$ .$V   T "  # % & *;DdM2$N >> % t~~ &Y' D t  r5cv|xsi}|tvr |jd|j|d<||i|S)Nr;)r7getr;)r=functypesargskwargss r6__torch_function__z DeviceContext.__torch_function__vsC2 ') )fjj.B.J#{{F8 T$V$$r5)returnN)r4N)__name__ __module__ __qualname__r>rFrMrTr4r5r6r9r9Ds4!$6%r5r9c"tfd|S)NcSr r4r;sr6z"device_decorator..sVr5r)r;rPs` r6device_decoratorr]~s ^T 22r5cfdS)z Set the default device inside of the wrapped function by decorating it with this function. If you would like to use this as a context manager, use device as a context manager directly, e.g., ``with torch.device(device)``. cBttj|Sr )r]r r;)rPr;s r6r\zset_device..s(f)=tDr5r4r[s`r6 set_devicer`s  EDr5) functoolsr torch._Crtorch.overridesrrrtorch.utils._contextlibrr r;__annotations__ lru_cacher7r9r]r`r4r5r6rgsg .DD5'+ t#*Q22l6%%6%t3Er5