o U)i@sUddlmZddlZddlmZmZmZddlmZddl m Z ddl Z da eej ed<e ddd ZGd d d eZd d ZddZdS))OptionalN)TorchFunctionMode _pop_mode _push_modecontext_decorator)_len_torch_function_stackCURRENT_DEVICEcCshtjtjtjtjtjtjtjtjtj tj j tj j tj tjtjtjtjtjtjtjjtjtjtjtjtjtjtjtjtjtjtjtj tj!tj"tj#tj$tj%tj&tj'tj(tj%SN))torchemptyZempty_permutedZ empty_stridedZempty_quantizedZonesZarangeZbartlett_windowZblackman_windoweyeZfftZfftfreqZrfftfreqfullfillZhamming_windowZ hann_windowZ kaiser_windowZlinspaceZlogspacenestedZ nested_tensorZrandZrandnrandintZrandpermrangeZsparse_coo_tensorZsparse_compressed_tensorZsparse_csr_tensorZsparse_csc_tensorZsparse_bsr_tensorZsparse_bsc_tensorZ tril_indicesZ triu_indicesZvanderZzerosZasarrayZtensorZ as_tensorZ scalar_tensorrr_/home/app/PaddleOCR-VL-test/.venv_paddleocr/lib/python3.10/site-packages/torch/utils/_device.py_device_constructors s      !"#$%&')*+,rc@s.eZdZddZddZddZd d d ZdS) DeviceContextcCst||_dSr )r device)selfrrrr__init__>szDeviceContext.__init__cCsBt|_|jaddttD}t|t|D]}t|qdS)NcSsg|]}tqSr)r).0_rrr Isz+DeviceContext.__enter__..)r old_devicerrrrreversed)r cur_stackmoderrr __enter__As  zDeviceContext.__enter__cCst|jag}ttdD]}t}t|trJ||q tdkr-t}t|ts-Jt|D]}t |q1dS)Nr r) rr rrr isinstancerappendrr)rexc_typeexc_valexc_tbr rr!rrr__exit__Qs    zDeviceContext.__exit__rNcCs8|pi}|tvr|ddur|j|d<||i|S)Nr)rgetr)rfunctypesargskwargsrrr__torch_function__ds z DeviceContext.__torch_function__)rN)__name__ __module__ __qualname__rr"r(r.rrrrr=s rcstfdd|S)NcsSr rrrrrlsz"device_decorator..r)rr*rr2rdevice_decoratorksr4cs fddS)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)``. cstt|Sr )r4r r)r*r2rrr3uszset_device..rr2rr2r set_devicens r5)typingrr Ztorch.overridesrrrZtorch.utils._contextlibrZtorch._Cr functoolsr r__annotations__ lru_cacherrr4r5rrrrs   1.