test_config.py

# A test on its own
import torch

# A test on its own
import deepspeed


def test_cuda():
    assert (torch.cuda.is_available())