You would need to do something like:
```python
class TestKVStoreGPU(unittest.TestCase):
@classmethod
def setUpClass(cls):
num_gpus = mx.context.num_gpus()
if num_gpus < 2:
raise unittest.SkipTest("It makes sense to test rsp_push_pull
functionality on more than 1 GPU only")
```[ Full content available at: https://github.com/apache/incubator-mxnet/pull/12726 ] This message was relayed via gitbox.apache.org for [email protected]
