GjUddlZddlZddlmZddlmZmZmZddlm Z da ej dze d<ej ddZGdd eZd Zd ZdS) N)_len_torch_function_stack) _pop_mode _push_modeTorchFunctionModecontext_decoratorCURRENT_DEVICEchtjtjtjtjtjtjtjtjtj tj j tj j tj tjtjtjtjtjtjjtjtjtjtjtjtjtjtjtjtjtjtj tj!tj"tj#tj$tj%tj&SN)'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_tensorV/var/www/html/Carbon-Document/venv/lib/python3.11/site-packages/torch/utils/_device.py_device_constructorsr7 s1  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*eZdZddZdZdZd dZdS) DeviceContextreturnNcFtj||_d|_dSr )r device prev_mode)selfr<s r6__init__zDeviceContext.__init__Esl6** /3r5ct|_|jadtt D}t |t |D].}t|tr||_ t |/dS)Nc*g|]}tSr4)r).0_s r6 z+DeviceContext.__enter__..QsMMMQY[[MMMr5) r old_devicer<r&rrreversed isinstancer9r=)r> cur_stackmodes r6 __enter__zDeviceContext.__enter__Is( NM%0I0K0K*L*LMMM 4Y'' ! !D$ .. !!%4     ! !r5c|jag}ttdz D]I}t }t |t rtd||Jtdkr2t }t |t std|j t|j t|D]}t|dS)Nr z@Found nested DeviceContext on the mode stack where none expectedrz8Expected a DeviceContext at the bottom of the mode stack) rEr r&rrrGr9AssertionErrorappendr=rrF)r>exc_typeexc_valexc_tbrHrCrIs r6__exit__zDeviceContext.__exit__[s  022Q677 # #A;;D$ .. $V   T " " " " $ & & * *;;DdM22 $N > % t~ & & &Y''  D t      r5r4cx|pi}|tvr|d |j|d<||i|S)Nr<)r7getr<)r>functypesargskwargss r6__torch_function__z DeviceContext.__torch_function__vsO2 ')) ) )fjj.B.B.J#{F8 tT$V$$$r5)r:N)r4N)__name__ __module__ __qualname__r?rJrQrXr4r5r6r9r9DsZ4444!!!$6%%%%%%r5r9c*tfd|S)NcSr r4r<sr6z"device_decorator..sVr5r)r<rTs` r6device_decoratorr`~s ^^^^T 2 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)``. cHttj|Sr )r`r r<)rTr<s r6r_zset_device..s(f)=)=tDDr5r4r^s`r6 set_devicercs E D D DDr5) functoolsr torch._Crtorch.overridesrrrtorch.utils._contextlibrr r<__annotations__ lru_cacher7r9r`rcr4r5r6rjs ......DDDDDDDDDD555555'+ t#***Q222l6%6%6%6%6%%6%6%6%t333EEEEEr5