˲jBUddlmZddlZddlmZmZmZddlmZddl m Z ddl Z da eej ed<e jddZGd d eZd Zd ZdS) )OptionalN)TorchFunctionMode _pop_mode _push_modecontext_decorator)_len_torch_function_stackCURRENT_DEVICEchtjtjtjtjtjtjtjtjtj tj j tj j tj tjtjtjtjtjtjtjjtjtjtjtjtjtjtjtjtjtjtjtj tj!tj"tj#tj$tj%tj&tj'tj(tj%SN))torchemptyempty_permuted empty_stridedempty_quantizedonesarangebartlett_windowblackman_windoweyefftfftfreqrfftfreqfullfillhamming_window hann_window kaiser_windowlinspacelogspacenested nested_tensorrandrandnrandintrandpermrangesparse_coo_tensorsparse_compressed_tensorsparse_csr_tensorsparse_csc_tensorsparse_bsr_tensorsparse_bsc_tensor tril_indices triu_indicesvanderzerosasarraytensor as_tensor scalar_tensorDC:\Projects\Historical\venv\Lib\site-packages\torch/utils/_device.py_device_constructorsr: s-  -  -   -   -   -   -  -  -   -   -   -   -   -  -  !- " #- $ %- & '- (  ")- .  /- 0  1- 2  3- 4  5- 6 7- 8  9- : ;- < &=- > ?- @ A- B C- D E- F G- H I- J  K- L  M- N  O- R  S- T U- V W- X  Y- -r8c(eZdZdZdZdZddZdS) DeviceContextc8tj||_dSr )rdevice)selfr>s r9__init__zDeviceContext.__init__>sl6** r8ct|_|jadtt D}t |t |D]}t |dS)Nc*g|]}tSr7)r).0_s r9 z+DeviceContext.__enter__..IsMMMQY[[MMMr8)r old_devicer>r(r rreversed)r? cur_stackmodes r9 __enter__zDeviceContext.__enter__Asu( NM%0I0K0K*L*LMMM 4Y''  D t      r8c~|jag}ttdz D]<}t }t |t rJ||=tdkr%t }t |t sJt|D]}t|dS)Nr r) rFr r(r r isinstancer<appendrGr)r?exc_typeexc_valexc_tbrHrDrIs r9__exit__zDeviceContext.__exit__Qs 022Q677 # #A;;D!$ 66 6 66   T " " " " $ & & * *;;DdM22 2 22Y''  D t      r8r7Ncx|pi}|tvr|d |j|d<||i|S)Nr>)r:getr>)r?functypesargskwargss r9__torch_function__z DeviceContext.__torch_function__dsO2 ')) ) )fjj.B.B.J#{F8 tT$V$$$r8)r7N)__name__ __module__ __qualname__r@rJrQrXr7r8r9r<r<=sU+++    &%%%%%%r8r<c*tfd|S)NcSr r7r>sr9z"device_decorator..lsVr8r)r>rTs` r9device_decoratorr`ks ^^^^T 2 22r8cfdS)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`rr>)rTr>s r9r_zset_device..us(f)=)=tDDr8r7r^s`r9 set_devicercns E D D DDr8)typingrrtorch.overridesrrrtorch.utils._contextlibrtorch._Cr functoolsr r>__annotations__ lru_cacher:r<r`rcr7r8r9rks  DDDDDDDDDD555555......)-&---Q...b+%+%+%+%+%%+%+%+%\333EEEEEr8