o
    Û·i"  ã                   @   sì   d dl mZmZmZmZmZmZmZmZm	Z	m
Z
mZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZmZm Z m!Z!m"Z"m#Z#m$Z$m%Z%m&Z&m'Z'm(Z(m)Z)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/ d dl0m1Z1m2Z2 d dl3m4Z4m5Z5m6Z6 dS )é   )/ÚDEFAULT_LAUNCH_COMMANDÚare_the_same_tensorsÚassert_exceptionÚcapture_call_outputÚdevice_countÚexecute_subprocess_asyncÚget_launch_commandÚget_torch_dist_unique_portÚmemory_allocated_funcÚpath_in_accelerate_packageÚpytest_xdist_worker_idÚrequire_bnbÚrequire_cpuÚrequire_cudaÚrequire_cuda_or_hpuÚrequire_cuda_or_xpuÚrequire_fp8Úrequire_fp16Úrequire_huggingface_suiteÚrequire_mluÚrequire_mpsÚrequire_multi_deviceÚrequire_multi_gpuÚrequire_multi_gpu_or_xpuÚrequire_multi_xpuÚrequire_musaÚrequire_non_cpuÚrequire_non_hpuÚrequire_non_torch_xlaÚrequire_non_xpuÚrequire_npuÚrequire_pippyÚrequire_sdaaÚrequire_single_deviceÚrequire_single_gpuÚrequire_single_xpuÚrequire_torch_min_versionÚrequire_torchaoÚrequire_torchvisionÚrequire_tpuÚrequire_transformer_engineÚ require_transformer_engine_mxfp8Úrequire_xpuÚ	run_firstÚskipÚslowÚtorch_device)ÚRegressionDatasetÚRegressionModel)Útest_scriptÚ	test_syncÚtest_opsN)7Útestingr   r   r   r   r   r   r   r	   r
   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r   r    r!   r"   r#   r$   r%   r&   r'   r(   r)   r*   r+   r,   r-   r.   r/   r0   Útrainingr1   r2   Úscriptsr3   r4   r5   © r9   r9   úT/home/ubuntu/vllm_env/lib/python3.10/site-packages/accelerate/test_utils/__init__.pyÚ<module>   s   Ä1