%ɷi)wddlZddlZddlZddlZddlZddlZddlZddlZddlZddl Z ddl m Z ddl m Z ddlmZddlmZmZddl mZddlZddlZddlmZdd lmZmZmZmZmZmZmZm 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/m0Z0m1Z1m2Z2m3Z3m4Z4m5Z5m6Z6m7Z7m8Z8m9Z9m:Z:m;Z;mZ>m?Z?m@Z@mAZAmBZBmCZCmDZDd ZEeE\ZFZGZHd eIfd ZJeJeGd ZKdedZLeLddZMdZNdZOdZPdZQdZRdZSdZTdZUdZVdZWdZXdZYdZZd Z[d!Z\d"Z]d#Z^d$Z_d%Z`d&Zad'Zbd(Zcd)Zdd*Zed+Zfd,Zgd-Zhd.Zid/Zjd0Zkd1Zld2Zmd3Znd4Zod5Zpd6Zqd7Zrdfd8Zsd9Ztd:Zud;Zvd<Zwd=Zxd>Zyd?Zzd@Z{dAZ|dBZ}dCZ~dDZdEZdFZdGZdHZeeBe6e=e5gxre ZdIZdJZdKZGdLdMe jZGdNdOe jZGdPdQe jZdRZGdSdTZdUZdgd efdVZdhdWeId efdXZdYZdZZGd[d\eZdid]eIefd^Zd_ed efd`Ze djdaedbeed efdcZddZy)kN)contextmanager)partial)Path)OptionalUnion)mock)AcceleratorState),check_cuda_fp8_capabilitycompare_versionsgatheris_aim_availableis_bnb_availableis_clearml_availableis_comet_ml_availableis_cuda_availableis_datasets_availableis_deepspeed_availableis_dvclive_availableis_fp8_availableis_fp16_availableis_habana_gaudi1is_hpu_availableis_import_timer_availableis_matplotlib_availableis_mlflow_availableis_mlu_availableis_mps_availableis_musa_availableis_neuron_availableis_npu_availableis_pandas_availableis_pippy_availableis_pytest_availableis_schedulefree_availableis_sdaa_availableis_swanlab_availableis_tensorboard_availableis_timm_availableis_torch_versionis_torch_xla_availableis_torchao_available*is_torchdata_stateful_dataloader_availableis_torchvision_availableis_trackio_availableis_transformer_engine_available%is_transformer_engine_mxfp8_availableis_transformers_availableis_triton_availableis_wandb_availableis_xpu_available str_to_boolc6tr9dtjjtjjfSt r9dtjjtjjfSt drddtjjfSt rdddfStr9dtjjtjjfStr9d tjjtjjfStr9d tjjtjjfStr9d tj jtj jfSt#r9d tj$jtj$jfSt'r9d tj(jtj(jfSt+r9dtj,jtj,jfSdddfS)Nxlacudaz2.0) min_versionmpscyNrr?P/var/www/html/venv/lib/python3.12/site-packages/accelerate/test_utils/testing.pyzget_backend..\r@mlusdaamusanpuxpuhpuneuroncpucyr>r?r?r@rArBzget_backend..lrCr@)r+torchr9 device_countmemory_allocatedrrr;current_allocated_memoryrrDr&rErrFr!rGr5rHrrIr rJr?r@rA get_backendrQTsejj--/1L1LLL  uzz..0%**2M2MMM e ,a;;;;  a""  eii,,. 0J0JJJ  uzz..0%**2M2MMM  uzz..0%**2M2MMM  eii,,. 0J0JJJ  eii,,. 0J0JJJ  eii,,. 0J0JJJ  224ell6S6SSSa""r@returnc ddg}|jD]F\}}t|tr|r|jd|-|0|jd|d|H|S)z Wraps around `kwargs` to help simplify launching from `subprocess`. Example: ```python # returns ['accelerate', 'launch', '--num_processes=2', '--device_count=2'] get_launch_command(num_processes=2, device_count=2) ``` acceleratelaunchz--=)items isinstanceboolappend)kwargscommandkvs rAget_launch_commandr_rsiX&G )1 a 1 NNRs8 $ ] NNRs!A3< ( ) Nr@g?) num_processesmonitor_intervalFc tj|} t|}|S#t$rtd|dwxYw#t$r|}Y|SwxYw)NzIf set, z must be yes or no.)osenvironr6 ValueErrorKeyError)keydefaultvalue_values rAparse_flag_from_envrksp B 3  B 'F M Bxu,?@A A B  Ms? #< A ARUN_SLOW)rhc8tjd|S)z+Decorator that skips a test unconditionallyzTest was skipped)unittestskip test_cases rAroros ,8==+ ,Y 77r@cBtjtd|S)z Decorator marking a test as slow. Slow tests are skipped by default. Set the RUN_SLOW environment variable to a truthy value to run them. z test is slow)rn skipUnless_run_slow_testsrps rAslowrus @8   ? JJr@cHtjtdk(d|S)zu Decorator marking a test that must be only ran on the CPU. These tests are skipped when a GPU is available. rKztest requires only a CPUrnrs torch_devicerps rA require_cpurys& R8  |u46P QR[ \\r@cHtjtdk7d|S)z Decorator marking a test that requires a hardware accelerator backend. These tests are skipped when there are no hardware accelerator available. rKtest requires a GPUrwrps rArequire_non_cpur|s% M8  |u46K LY WWr@cdtjtxr t d|S)z Decorator marking a test that requires CUDA. These tests are skipped when there are no GPU available or when TorchXLA is available. r{)rnrsrr+rps rA require_cudar~s1 l8  02S;Q;S7SUj klu vvr@c|tjtxr t xs t d|S)z Decorator marking a test that requires CUDA or HPU. These tests are skipped when there are no GPU available or when TorchXLA is available. ztest requires a GPU or HPU)rnrsrr+rrps rArequire_cuda_or_hpurs?  8     =%;%=!=TBRBTVr  r@cJtjtd|S)zn Decorator marking a test that requires XPU. These tests are skipped when there are no XPU available. test requires a XPU)rnrsr5rps rA require_xpurs$ J8  /13H I) TTr@ctxr t }t}tj|xs|d|S)z Decorator marking a test that requires CUDA or XPU. These tests are skipped when there are no GPU available or when TorchXLA is available. ztest requires a CUDA GPU or XPU)rr+r5rnrs)rqcuda_condition xpu_conditions rArequire_cuda_or_xpursA '(I1G1I-IN$&M b8  ~>@a bcl mmr@cHtjtdk7d|S)zB Decorator marking a test that should be skipped for XPU. rHztest requires a non-XPUrwrps rArequire_non_xpur& Q8  |u46O PQZ [[r@cHtjtdk7d|S)zB Decorator marking a test that should be skipped for HPU. rIztest requires a non-HPUrwrps rArequire_non_hpurrr@cJtjtd|S)zj Decorator marking a test that requires FP16. These tests are skipped when FP16 is not supported. ztest requires FP16 support)rnrsrrps rA require_fp16rs% R8  024P QR[ \\r@ct}tjjr t sd}t r t rd}tj|d|S)zh Decorator marking a test that requires FP8. These tests are skipped when FP8 is not supported. Fztest requires FP8 support) rrMr9 is_availabler rrrnrs)rqfp8_is_availables rA require_fp8rsW() zz )B)D .0  M8  /1L Mi XXr@cNtjtddd|S)N>=z2.5.0z$test requires FSDP2 (torch >= 2.5.0))rnrsr*rps rA require_fsdp2rs& g8  /g>@f ghq rrr@cJtjtd|S)zn Decorator marking a test that requires MLU. These tests are skipped when there are no MLU available. ztest require a MLU)rnrsrrps rA require_mlur$ I8  /13G H SSr@cJtjtd|S)zp Decorator marking a test that requires SDAA. These tests are skipped when there are no SDAA available. ztest require a SDAA)rnrsr&rps rA require_sdaar $ K8  024I J9 UUr@cJtjtd|S)zp Decorator marking a test that requires MUSA. These tests are skipped when there are no MUSA available. ztest require a MUSA)rnrsrrps rA require_musarrr@cJtjtd|S)zn Decorator marking a test that requires NPU. These tests are skipped when there are no NPU available. ztest require a NPU)rnrsr!rps rA require_npurrr@cJtjtd|S)zz Decorator marking a test that requires Neuron. These tests are skipped when there are no Neuron Cores available. ztest require Neuron Cores)rnrsr rps rArequire_neuronr"s% S8  246Q RS\ ]]r@cJtjtd|S)z Decorator marking a test that requires MPS backend. These tests are skipped when torch doesn't support `mps` backend. z0test requires a `mps` backend support in `torch`)rnrsrrps rA require_mpsr)s% g8  /13e fgp qqr@cbtjtxr td|S)zv Decorator marking a test that requires transformers and datasets. These tests are skipped when they are not. z$test requires the Hugging Face suite)rnrsr2rrps rArequire_huggingface_suiter1s5 8  !#?(=(?.  r@cJtjtd|S)zi Decorator marking a test that requires transformers. These tests are skipped when they are not. z&test requires the transformers library)rnrsr2rps rArequire_transformersr;% f8  8:S TU^ __r@cHtjtdkDd|S)z Decorator marking a test that requires a multi-device setup. These tests are skipped on a machine without multiple devices. r<z,test requires multiple hardware accelerators)rnrsrNrps rArequire_multi_devicers& a8  |a/1_ `aj kkr@cxtjtjj dkDd|S)z Decorator marking a test that requires a multi-GPU setup. These tests are skipped on a machine without multiple GPUs. r<ztest requires multiple GPUsrrps rArequire_multi_gpurs3 ]8  uzz6681<>[ \]f ggr@cxtjtjj dkDd|S)z Decorator marking a test that requires a multi-XPU setup. These tests are skipped on a machine without multiple XPUs. r<ztest requires multiple XPUsrrps rArequire_multi_xpurs3 \8  uyy557!;=Z [\e ffr@cxtjtxs txr tdkDd|S)z Decorator marking a test that requires a multi-GPU setup. These tests are skipped on a machine without multiple GPUs or XPUs. r<z#test requires multiple GPUs or XPUs)rnrsrr5rNrps rArequire_multi_gpu_or_xpurs=  8     2 0 2H q8HJo  r@cJtjtd|S)z| Decorator marking a test that requires DeepSpeed installed. These tests are skipped when DeepSpeed isn't installed ztest requires DeepSpeed)rnrsrrps rArequire_deepspeedrs% T8  579R ST] ^^r@cltjtddxr tdddd|S)zn Decorator marking a test that requires TP installed. These tests are skipped when TP isn't installed rz2.3.0 transformersz4.52.0zGtest requires torch version >= 2.3.0 and transformers version >= 4.52.0)rnrsr*r rps rA require_tprs@ 8  w'\,<^TS[,\Q  r@cz|tt|Stjt d|d||S)z Decorator marking that a test requires a particular torch version to be tested. These tests are skipped when an installed torch version is less than the required one. )versionrztest requires torch version >= )rrequire_torch_min_versionrnrsr*)rqrs rArrsE 0'BB l8  /g>Babiaj@k lmv wwr@cJtjtd|S)z Decorator marking a test that requires tensorboard installed. These tests are skipped when tensorboard isn't installed ztest requires Tensorboard)rnrsr(rps rArequire_tensorboardrs% X8  79;V WXa bbr@cJtjtd|S)zt Decorator marking a test that requires wandb installed. These tests are skipped when wandb isn't installed ztest requires wandb)rnrsr4rps rA require_wandbrs$ L8  135J KI VVr@cJtjtd|S)zx Decorator marking a test that requires trackio installed. These tests are skipped when trackio isn't installed ztest requires trackio)rnrsr/rps rArequire_trackior% P8  357N OPY ZZr@cJtjtd|S)zz Decorator marking a test that requires comet_ml installed. These tests are skipped when comet_ml isn't installed ztest requires comet_ml)rnrsrrps rArequire_comet_mlrs% R8  468P QR[ \\r@cJtjtd|S)zp Decorator marking a test that requires aim installed. These tests are skipped when aim isn't installed ztest requires aim)rnrsrrps rA require_aimrs$ H8  /13F G RRr@cJtjtd|S)zx Decorator marking a test that requires clearml installed. These tests are skipped when clearml isn't installed ztest requires clearml)rnrsrrps rArequire_clearmlrrr@cJtjtd|S)zx Decorator marking a test that requires dvclive installed. These tests are skipped when dvclive isn't installed ztest requires dvclive)rnrsrrps rArequire_dvcliverrr@cJtjtd|S)zx Decorator marking a test that requires swanlab installed. These tests are skipped when swanlab isn't installed ztest requires swanlab)rnrsr'rps rArequire_swanlabrrr@cJtjtd|S)zv Decorator marking a test that requires pandas installed. These tests are skipped when pandas isn't installed ztest requires pandas)rnrsr"rps rArequire_pandasr$ N8  246L Mi XXr@cJtjtd|S)zv Decorator marking a test that requires mlflow installed. These tests are skipped when mlflow isn't installed ztest requires mlflow)rnrsrrps rArequire_mlflowr rr@cdtjtxr t d|S)z Decorator marking a test that requires pippy installed. These tests are skipped when pippy isn't installed It is also checked if the test is running on a Gaudi1 device which doesn't support pippy. ztest requires pippy)rnrsr#rrps rA require_pippyrs1 g8  13N@Be fgp qqr@cJtjtd|S)z Decorator marking a test that requires transformers engine MXFP8 block scaling available. These tests are skipped when transformers engine MXFP8 block scaling isn't available z5test requires transformers engine MXFP8 block scaling)rnrsr1rps rA require_transformer_engine_mxfp8r*s*  8  -/1h  r@cJtjtd|S)zx Decorator marking a test that requires torchao installed. These tests are skipped when torchao isn't installed ztest requires torchao)rnrsr,rps rArequire_torchaor4rr@cJtjtd|S)z Decorator marking a test that requires matplotlib installed. These tests are skipped when matplotlib isn't installed ztest requires matplotlib)rnrsrrps rArequire_matplotlibr;s% V8  68:T UV_ ``r@cBtjtd|S)z Decorator marking that a test requires at least one tracking library installed. These tests are skipped when none are installed zYtest requires at least one tracker to be available and for `comet_ml` to not be installed)rnrs_atleast_one_tracker_availablerps rArequire_trackersrIs)  8  &c  r@cJtjtd|S)z Decorator marking a test that requires torchdata.stateful_dataloader. These tests are skipped when torchdata with stateful_dataloader module isn't installed. z+test requires torchdata.stateful_dataloader)rnrsr-rps rA%require_torchdata_stateful_dataloaderrTs* 8  246c  r@cdtr%ddl}|jjd|S|S)a Decorator marking a test with order(1). When pytest-order plugin is installed, tests marked with this decorator are guaranteed to run first. This is especially useful in some test settings like on a Gaudi instance where a Gaudi device can only be used by a single process at a time. So we make sure all tests that run in a subprocess are launched first, to avoid device allocation conflicts. If pytest is not installed, test will be returned as is. rNr<)r$pytestmarkorder)rqrs rA run_firstr`s/#v{{  #I.. r@c:eZdZdZdZedZedZdZy)TempDirTestCasea A TestCase class that keeps a single `tempfile.TemporaryDirectory` open for the duration of the class, wipes its data at the start of a test, and then destroys it at the end of the TestCase. Useful for when a class or API requires a single constant folder throughout it's use, such as Weights and Biases The temporary directory location will be stored in `self.tmpdir` TcHttj|_y)zECreates a `tempfile.TemporaryDirectory` and stores it in `cls.tmpdir`N)rtempfilemkdtemptmpdirclss rA setUpClasszTempDirTestCase.setUpClasss(**,- r@ctjj|jr t j |jyy)z1Remove `cls.tmpdir` after test suite has finishedN)rcpathexistsrshutilrmtreers rA tearDownClasszTempDirTestCase.tearDownClasss- 77>>#** % MM#** % &r@c|jrh|jjdD]I}|jr|j $|j s5t j|Kyy)zKKM[[]MM$'  ( r@N) __name__ __module__ __qualname____doc__r  classmethodrr rr?r@rArrss9N..&& (r@rc"eZdZdZfdZxZS)AccelerateTestCasez A TestCase class that will reset the accelerator state at the end of every test. Every test that checks or utilizes the `AcceleratorState` class should inherit from this to avoid silent failures due to state being shared between tests. cLt|tjdy)NT)supertearDownr _reset_state)r __class__s rArzAccelerateTestCase.tearDowns %%d+r@)rrrrr __classcell__)rs@rArrs ,,r@rcTeZdZdZdeej eej ffdZy)MockingTestCasea A TestCase class designed to dynamically add various mockers that should be used in every test, mimicking the behavior of a class-wide mock when defining one normally will not do. Useful when a mock requires specific information available only initialized after `TestCase.setUpClass`, such as setting an environment variable with that information. The `add_mocks` function should be ran at the end of a `TestCase`'s `setUp` function, after a call to `super().setUp()` such as: ```python def setUp(self): super().setUp() mocks = mock.patch.dict(os.environ, {"SOME_ENV_VAR", "SOME_VALUE"}) self.add_mocks(mocks) ``` mocksct|ttfr|n|g|_|jD]-}|j |j |j /y)aQ Add custom mocks for tests that should be repeated on each test. Should be called during `MockingTestCase.setUp`, after `super().setUp()`. Args: mocks (`mock.Mock` or list of `mock.Mock`): Mocks that should be added to the `TestCase` after `TestCase.setUpClass` has been run N)rXtuplelistr#start addCleanupstop)rr#ms rA add_mockszMockingTestCase.add_mockssH) >UUG  $A GGI OOAFF # $r@N) rrrrrrMockr&r+r?r@rAr"r"s)" $uTYYTYY%?@ $r@r"c8t}|djj|j}t |j }|dj }t |jdD]}tj|||ryy)NrFT) r clonetodevicer rKrangeshaperMequal)tensorstatetensorsis rAare_the_same_tensorsr8s  E D\   ! $ $U\\ 2FVn  "G AY]]_F 7==# ${{71:v. r@ceZdZdZy) _RunOutputc.||_||_||_yN) returncodestdoutstderr)rr=r>r?s rA__init__z_RunOutput.__init__s$  r@N)rrrr@r?r@rAr:r:sr@r:cZK |jd{}|r ||ny%7wr<)readline)streamcallbacklines rA _read_streamrFs- __&&  TN  &s +)+c T K|rtddj|tj|dg|dd|tjj tjj |dd{}ggd fd tj tjt|j fdtjt|j fdg| d{t|j d{S77)7w) Nz Running:  rr<)stdinr>r?envc|jdj}|j|st|||yy)Nutf-8)file)decoderstriprZprint)rEsinkpipelabelquiets rAteez_stream_subprocess..tees:{{7#**, D %D )r@c8|tjdS)Nzstdout:rS)sysr>)loutrUs rArBz$_stream_subprocess..QSZZ_hAir@c8|tjdS)Nzstderr:rW)rXr?)rYerrrUs rArBz$_stream_subprocess..r[r@)timeout)) rPjoinasynciocreate_subprocess_exec subprocessPIPEwait create_taskrFr>r?r:) cmdrJrIr^rTechopr]rZrUs ` @@@rA_stream_subprocessrjs  mSXXc]+,, A QR!!&&!!&&   A" C C* ,,    QXX7i j k    QXX7i j k    AFFHnc3 //E 6%s7A3D(9D":BD(:D$;D(D& D($D(&D(rgc dt|D]$\}}t|tst|||<&t j t ||||||}dj|} |jdkDr9dj|j} td| d|jd| |S)N)rJrIr^rTrhrHr 'z' failed with returncode z, The combined stderr from workers follows: ) enumeraterXrstrrarunrjr`r=r? RuntimeError) rgrJrIr^rTrhr7cresultcmd_strr?s rAexecute_subprocess_asyncrus#1 a VCF[[+CSw^cjno pFhhsmG 16==)y1&2C2C1DE::@ C  Mr@ctjjdd}tjdd|dtj }t |S)z Returns an int value of worker's numerical id under `pytest-xdist`'s concurrent workers `pytest -n N` regime, or 0 if `-n 1` or `pytest-xdist` isn't being used. PYTEST_XDIST_WORKERgw0z^gwr_r)rcrdgetresubMint)workers rApytest_xdist_worker_idrs= ZZ^^15 9F VVFB244 0F v;r@c$d}t}||zS)z Returns a port number that can be fed to `torch.distributed.launch`'s `--master_port` argument. Under `pytest-xdist` it adds a delta number based on a worker id so that concurrent tests don't try to use the same port at once. i}tddj|d|jj|d}~wwxYw)z Runs `command` with `subprocess.check_output` and will potentially return the `stdout`. Will also properly capture if an error occurred while running `command` N)r?rJrNrLz Command `rHz$` failed with the following error: )rnrXrrorcrdcopyrc check_outputSTDOUThasattrrNCalledProcessErrorrr`output)r\ return_stdoutrJr7rrres rA run_commandr3s '" 1 a QGAJ  {jjoo ((9J9JPST vx(w/M   ( (%)**PQRQYQYQ`Q`QbPc d  sABC-/9C((C- componentscxttjtj}|j |S)z Get a path within the `accelerate` package's directory. Args: *components: Components of the path to join after the package directory. Returns: `Path`: The path to the requested file or directory. )rinspectgetfilerTparentjoinpath)raccelerate_package_dirs rApath_in_accelerate_packagerJs2"'//*"=>EE * ! * *J 77r@exception_classmsgc#Kd} dd}|rt d|d y#t$rR}t||sJd|dt||$|t|vsJd|dt|dYd}~hd}~wwxYww) z Context manager to assert that the right `Exception` class was raised. If `msg` is provided, will check that the message is contained in the raised exception. FNTzExpected exception of type z but got zExpected message 'z' to be in exception but got 'rmz but ran without issue.) ExceptionrXtyperoAssertionError)rrwas_ranrs rAassert_exceptionrYsGd  :?:KKbcdd d!_-p1L_L]]fgklmgnfo/pp- ?#a&= c$6se;YZ]^_Z`Yaab"c c=ds+A<A< A9AA4/A<4A99A<ctj}tj} |t_||i| |t_|j S#t$r}|d}~wwxYw#|t_wxYw)z` Takes in a `func` with `args` and `kwargs` and returns the captured stdout as a string N)ioStringIOrXr>rgetvalue)funcargsr[captured_outputoriginal_stdoutrs rAcapture_call_outputrlsjkkmOjjO%$  df%  # # %% % s#A A%A  A%%A(( A5)F)NN)NNNFF)NNFT)FNr<)rarrrcrzr rcrXrrn contextlibr functoolsrpathlibrtypingrrrrMrTr5r utilsr r r rrrrrrrrrrrrrrrrrrr r!r"r#r$r%r&r'r(r)r*r+r,r-r.r/r0r1r2r3r4r5r6rQrxrNmemory_allocated_funcr&r_DEFAULT_LAUNCH_COMMANDrkrtroruryr|r~rrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrrranyrrrrTestCaserrr"r8r:rFrjrurrrrrorrrYrrr?r@rArs   %" $------------`#65@M1 l1D&,,Y\]  &j%@8 K]XwUn\\]Y$sTVVT^rqaoeqhjqa`lhg_xcW[]S[[[YYrir[a  79;O;QSgSijk$ ! ##  &(h''(B ,** ,$h''$B&0bl&0R$fp&  i c. 8C 8D 8eiehsmetee$ &r@