code: def _skip_on_cuda_compatability():
      if torch.cuda.get_device_capability()[0] < 7:
          pytest.skip("needs higher compute capability than 7")
      cuda_major = int(torch.version.cuda.split(".")[0]) * 10
      cuda_minor = int(torch.version.cuda.split(".")[1])
      cuda_version = cuda_major + cuda_minor
      if (cuda_version != 101 and cuda_version != 102) and (
          cuda_version != 111 and cuda_version != 110
      ):
          pytest.skip("requires cuda 10.1 or 10.2 or 11.0 or 11.1")
  
file_path: H2O/h2o_flexgen/benchmark/third_party/DeepSpeed/tests/unit/ops/sparse_attention/test_sparse_attention.py
imports: import pytest
  import torch
  import deepspeed
  from deepspeed.ops.op_builder import SparseAttnBuilder
label_desc: Use image node_id 8 for calling a global function with example usage: _skip_on_cuda_compatability() without return types