o i@sXddlZddlmZmZer ddlZeeddZddZnddZ e Zd d Z dS) N) is_availablecheck_availableCUPY_TEST_GPU_LIMITz-1cOstjj|i|S)N)pytestmarkslowargskwargsr O/home/ubuntu/veenaModal/venv/lib/python3.10/site-packages/cupy/testing/_attr.pyr srcOs tdJ)Nzpytest attributesrrr 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) oscupy.testing._pytest_implrrrintgetenvrrrrr r r r s