a NSic5ã@sJddlZddlmZddlmZd eeeedœdd„Zee dœd d „Z dS) éN)ÚAny)Ú_get_device_indexF)ÚdeviceÚoptionalÚ allow_cpuÚreturncCs€t|tƒrt |¡}t|tjƒrV|r>|jdvrVtd |¡ƒ‚n|jdkrVtd |¡ƒ‚tj ¡stt|tj jƒrt|j St |||ƒS)a¶Gets the device index from :attr:`device`, which can be a torch.device object, a Python integer, or ``None``. If :attr:`device` is a torch.device object, returns the device index if it is a CUDA device. Note that for a CUDA device without a specified index, i.e., ``torch.device('cuda')``, this will return the current default CUDA device if :attr:`optional` is ``True``. If :attr:`allow_cpu` is ``True``, CPU devices will be accepted and ``-1`` will be returned in this case. If :attr:`device` is a Python integer, it is returned as is. If :attr:`device` is ``None``, this will return the current default CUDA device if :attr:`optional` is ``True``. )ÚcudaÚcpuz*Expected a cuda or cpu device, but got: {}rz#Expected a cuda device, but got: {}) Ú isinstanceÚstrÚtorchrÚtypeÚ ValueErrorÚformatÚjitÚ is_scriptingrÚidxÚ_torch_get_device_index)rrr©rúM/var/www/html/django/DPS/env/lib/python3.9/site-packages/torch/cuda/_utils.pyrs      r)ÚnamercCsdd„}t|tfd|iƒS)NcSs|jj}td |¡ƒ‚dS)Nz(Tried to instantiate dummy base class {})Ú __class__Ú__name__Ú RuntimeErrorr)ÚselfÚ class_namerrrÚinit_err&sÿz_dummy_type..init_errÚ__init__)r Úobject)rrrrrÚ _dummy_type%sr)FF) r ÚtypingrZ torch._utilsrrÚboolÚintr r rrrrrÚs  ÿÿ