!Ei= DSSKJr SSKrSSKJr S S\S\S\S\4Sjjrg) )AnyN)_get_device_indexdeviceoptional allow_cpureturnc[U[5(aU$[U[5(a[R"U5n[U[R5(aDU(aUR S;a[ SU35eOUR S:wa[ SU35e[RR5(d5[U[RR5(a UR$[XU5$)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 MTIA device. Note that for a MTIA device without a specified index, i.e., ``torch.device('mtia')``, this will return the current default MTIA 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 MTIA device if :attr:`optional` is ``True``. )mtiacpuz(Expected a mtia or cpu device, but got: r z!Expected a mtia device, but got: ) isinstanceintstrtorchrtype ValueErrorjit is_scriptingr idx_torch_get_device_index)rrrs N/home/mande/repo/quber/.venv/lib/python3.13/site-packages/torch/mtia/_utils.pyrr s &# &#f%&%,,'' {{/1 #KF8!TUU2 [[F "@IJ J 99 ! ! # # fejj// 0 0::  "6Y ??)FF)typingrr torch._utilsrrboolr rrrsA F