Ë
    §Ùbi  ã                   óì   — 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. d dl/m0Z0m1Z1m2Z2 d dl3m4Z4m5Z5m6Z6 y)é   ).Ú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_xpuÚ	run_firstÚskipÚslowÚtorch_device)ÚRegressionDatasetÚRegressionModelÚRegressionModel4XPU)Ú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/   r0   Útrainingr1   r2   r3   Úscriptsr4   r5   r6   © ó    úY/home/cdr/jupyterlab/.venv/lib/python3.12/site-packages/accelerate/test_utils/__init__.pyú<module>r=      sS   ð÷/÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷` NÑ M÷ 6Ò 5r;   