Vh:ddlmZddlZddlmZddededefdZy) )OptionalN)Devicedeviceoptionalreturnct|tr|St|trtj|}d}t|tjrtj j }| td|j|jk7r7t|jdtj j d|j}|.|std|tj jS|S)NzAccelerator expectedz' doesn't match the current accelerator .zFExpected a torch.device with a specified index or an integer, but got:) isinstanceintstrtorchr acceleratorcurrent_accelerator RuntimeErrortype ValueErrorindexcurrent_device_index)rr device_indexaccs H/home/dcms/DCMS/lib/python3.12/site-packages/torch/accelerator/_utils.py_get_device_indexrs&# &#f%"&L&%,,'335 ;56 6 88v{{ ";;-FuGXGXGlGlGnFoopq || XY_X`a   5577 )F) typingrr torch.typesr _device_tboolr rrrrs( +i4Cr