o 9wir@s ddlZddlZddlZddlmZmZmZmZddlZddl m Z m Z ddl m Z ddlmZddlmZddlmZddlmZdd lmZddddd d dd d d d d d ed eedeedeedededeedededededeeeeeefffddZdS)N)DictListOptionalTuple)RequirementCachecompare_version)Version)XLAAccelerator)num_cuda_devices)MPSAccelerator)_DEEPSPEED_AVAILABLE)_TORCH_GREATER_EQUAL_2_4F) min_cuda_gpus min_torch max_torch min_python bf16_cudatpumps skip_windows standalone deepspeeddynamorrrrrrrrrrrreturnc  Cs*g} i} |rt|kr| d|d| d<|r/tdtj|ddr/| d|dtjd|rGtdtj|ddrG| d |dtjd|ritj j d tj j d tj j } t | t |kri| d ||rz tjoutj }Wn&ttfy}zd t|vpd t|v}|r|d}WYd}~nd}~ww|r| d|rtjdkr| d|rts| dd| d<|dur|rts| dn |str| d|rtdddkr| dd| d<| rtrtstdds| d| rddlm}|s| d | | fS)!aConstruct reasons for pytest skipif. Args: min_cuda_gpus: Require this number of gpus and that the ``PL_RUN_CUDA_TESTS=1`` environment variable is set. min_torch: Require that PyTorch is greater or equal than this version. max_torch: Require that PyTorch is less than this version. min_python: Require that Python is greater or equal than this version. bf16_cuda: Require that CUDA device supports bf16. tpu: Require that TPU is available. mps: If True: Require that MPS (Apple Silicon) is available, if False: Explicitly Require that MPS is not available skip_windows: Skip for Windows platform. standalone: Mark the test as standalone, our CI will run it in a separate process. This requires that the ``PL_RUN_STANDALONE_TESTS=1`` environment variable is set. deepspeed: Require that microsoft/DeepSpeed is installed. dynamo: Require that `torch.dynamo` is supported. zGPUs>=Trtorch)use_base_versionztorch>=z, z installedztorch<.zpython>=zFound no NVIDIA driverzTorch not compiled with CUDANzCUDA device bf16win32zunimplemented on WindowsTPUrMPSznot MPSPL_RUN_STANDALONE_TESTS01zStandalone executionrzdeepspeed.utils)module Deepspeedr)is_dynamo_supportedz torch.dynamo)r appendroperatorltr __version__gesys version_infomajorminormicrorcuda is_availableis_bf16_supportedAssertionError RuntimeErrorstrplatformr r osgetenvr r rtorch._dynamo.eval_framer%)rrrrrrrrrrrreasonskwargs py_versioncondex is_unrelatedr%r@f/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/lightning/fabric/utilities/testing/_runif.py_runif_reasonssl              rB)r'r7r+typingrrrrr lightning_utilities.core.importsrrpackaging.versionrlightning.fabric.acceleratorsr "lightning.fabric.accelerators.cudar !lightning.fabric.accelerators.mpsr %lightning.fabric.strategies.deepspeedr "lightning.fabric.utilities.importsr intr5boolrBr@r@r@rAs`