Python get_resources_for_backends примеры использования

Пример #1

def test_common_backends_compute_2():
    resources = get_resources_for_backends(
        [('numpy', 'cupy'), (
            'numpy',
            'cupy',
        )],
        user_backends=('numpy', 'cupy'),
    )
    assert resources == {'compute': 1, 'ndarray': 1}

Пример #2

def test_common_backends_as_string():
    resources = get_resources_for_backends(
        ['cupy', ('cupy', )],
        user_backends='cupy',
    )
    assert resources == {'CUDA': 1, 'compute': 1, 'ndarray': 1}

Пример #3

def test_common_backends_gpu_3():
    resources = get_resources_for_backends(
        [('cupy', ), ('cupy', )],
        user_backends=('cupy', ),
    )
    assert resources == {'CUDA': 1, 'compute': 1, 'ndarray': 1}

Пример #4

def test_no_common_backends():
    with pytest.raises(ValueError):
        get_resources_for_backends(
            [('numpy', ), ('cupy', )],
            user_backends=None,
        )