§
    ¨g  ã                   óÂ   — 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# d dl$m%Z%m&Z&m'Z' d dl(m)Z)m*Z*m+Z+ dS )é   )#ÚDEFAULT_LAUNCH_COMMANDÚare_the_same_tensorsÚassert_exceptionÚcapture_call_outputÚdevice_countÚexecute_subprocess_asyncÚget_launch_commandÚmemory_allocated_funcÚpath_in_accelerate_packageÚrequire_bnbÚrequire_cpuÚrequire_cudaÚrequire_huggingface_suiteÚrequire_mluÚrequire_mpsÚrequire_multi_deviceÚrequire_multi_gpuÚrequire_multi_xpuÚrequire_musaÚrequire_non_cpuÚrequire_non_torch_xlaÚrequire_non_xpuÚrequire_npuÚrequire_pippyÚrequire_single_deviceÚrequire_single_gpuÚrequire_single_xpuÚrequire_torch_min_versionÚrequire_torchvisionÚrequire_tpuÚrequire_transformer_engineÚrequire_xpuÚskipÚslowÚtorch_device)ÚRegressionDatasetÚRegressionModelÚRegressionModel4XPU)Útest_scriptÚ	test_syncÚtest_opsN),Ú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%   Útrainingr&   r'   r(   Úscriptsr)   r*   r+   © ó    úZ/var/www/html/ai-engine/env/lib/python3.11/site-packages/accelerate/test_utils/__init__.pyú<module>r2      s¿  ðð$ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ð $ðJ NÐ MÐ MÐ MÐ MÐ MÐ MÐ MÐ MÐ Mð 6Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5Ð 5r0   