3j| 8SSKJr SSKrSS\S\S\S\4Sjjrg) )AnyNdeviceoptional allow_cpureturncUc%U(a[RR5$[U[5(aU$[R R 5(d5[U[RR5(a UR$[U[5(a[R"U5nSn[U[R5(aaU(dURS:Xa[SU35eURS;a[SU35eURS:XaSO URnUc5U(a [RR5nU$[SU35eU$)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``. Ncpuz$Expected a non cpu device, but got: )mtiar z(Expected a mtia or cpu device, but got: zGExpected a torch.device with a specified index or an integer, but got: )torch_C_mtia_getDevice isinstanceintjit is_scriptingr ridxstrtype ValueErrorindex)rrr device_idxs K/home/wildlama/miniconda3/lib/python3.13/site-packages/torch/mtia/_utils.py_get_device_indexrs1"~(xx''))&# 99 ! ! # # fejj// 0 0:: &#f%!J&%,,''V[[E1CF8LM M ;;o -GxPQ Q!;;%/RV\\  113J YZ`Yab  )FF)typingrr boolrrrrrs6