ó
    /ýòh  ã                   óì   • S SK JrJrJrJrJrJrJrJrJ	r	J
r
JrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJrJ r J!r!J"r"J#r#J$r$J%r%J&r&J'r'J(r(J)r)J*r*J+r+J,r,J-r-J.r.  S SK/J0r0J1r1J2r2  S SK3J4r4J5r5J6r6  g)é   ).Ú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   © ó    ÚX/home/james-whalen/.local/lib/python3.13/site-packages/accelerate/test_utils/__init__.pyÚ<module>r=      sS   ð÷/÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷ /÷` NÑ M÷ 6Ò 5r;   