tj7 BddlmZddlZddlmZ d dedededefd ZdS) )AnyN)_get_device_indexFdeviceoptional allow_cpureturnct|tr|St|trtj|}t|tjr;|r|jdvrt d|n|jdkrt d|tjs&t|tj jr|j St|||S)aGet 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 XPU device. Note that for a XPU device without a specified index, i.e., ``torch.device('xpu')``, this will return the current default XPU 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 XPU device if :attr:`optional` is ``True``. )xpucpuz'Expected a xpu or cpu device, but got: r z Expected a xpu device, but got: ) isinstanceintstrtorchrtype ValueErrorjit is_scriptingr idx_torch_get_device_index)rrrs N/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/torch/xpu/_utils.pyrr s"&# &#&f%%&%,''J  J{.00 !S6!S!STTT1 [E ! !HHHII I 9 ! ! # # fei. / / :  "68Y ? ??)FF)typingrr torch._utilsrrboolr rrrs FEEEEE