o X۷i@sdddlmZddlZddlmZmZer&ddlZeeddZ ddZ ndd Z e Z d d Z dS) ) annotationsN) is_availablecheck_availableCUPY_TEST_GPU_LIMITz-1cOstjj|i|S)N)pytestmarkslowargskwargsr H/home/ubuntu/vllm_env/lib/python3.10/site-packages/cupy/testing/_attr.pyrsrcOs tdJ)Nzpytest attributesrr r r r _dummy_callablesrcs$tddks Jfdd}|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 attributecs6tjjdtko knddtj|S)Nrz{} GPUs required)reason)rrskipif _gpu_limitformat multi_gpu)fgpu_numr r _wrapper(s  zmulti_gpu.._wrapperr)rrr rr rs  r) __future__roscupy.testing._pytest_implrrrintgetenvrrrrr r r r s