o ۷i9@s2ddlZddlmZddZedkredSdS)N)is_xpu_availablecCsPd}d}tjrtj}d}n trtj}d}td|d|ddS)NGPUrXPUzSuccessfully ran on  s)torchcuda is_available device_countrxpuprint)accelerator_typenum_acceleratorsr\/home/ubuntu/vllm_env/lib/python3.10/site-packages/accelerate/test_utils/scripts/test_cli.pymains   r__main__)raccelerate.utilsrr__name__rrrrs