`iddlZddlmZmZer"ddlZeejddZdZndZ e ZdZ dS)N) is_availablecheck_availableCUPY_TEST_GPU_LIMITz-1c.tjj|i|S)N)pytestmarkslowargskwargss f/home/jaya/work/projects/VOICE-AGENT/VIET/agent-env/lib/python3.11/site-packages/cupy/testing/_attr.pyr r s{0000c$tdJ)Nzpytest attributesrr s r _dummy_callablers+,,,urc@tddksJfd}|S)a{Decorator to indicate number of GPUs required to run the test. Tests can be annotated with this decorator (e.g., ``@multi_gpu(2)``) to declare number of GPUs required to run. When running tests, if ``CUPY_TEST_GPU_LIMIT`` environment variable is set to value greater than or equals to 0, test cases that require GPUs more than the limit will be skipped. zmulti_gpu attributectjdtcxkokncdtj|S)Nrz{} GPUs required)reason)rrskipif _gpu_limitformat multi_gpu)fgpu_nums r _wrapperzmulti_gpu.._wrapper&sq{!!  % % % %g % % % %%,,W55"77 %%a((** *rr)rrs` r rrsC)*** w;;;;***** Or) oscupy.testing._pytest_implrrrintgetenvrr rrrr r"s DCCCCCCC<>> MMMYRY4d;;<