Python load_backendの例

コード例 #1

def backend_opt(request):
    name, options = request.param
    try:
        b = load_backend(name, options)
    except LoadingError as e:
        pytest.skip(f"Can't load {name}: {e.__cause__}")
    return b

コード例 #2

ファイル: test_relay.py プロジェクト: xyz3021/myia

def test_relay_backend_bad_exec():
    with pytest.raises(Exception):
        load_backend("relay", dict(target="cpu", exec_kind="potato"))

コード例 #3

ファイル: test_relay.py プロジェクト: xyz3021/myia

def test_relay_backend_bad_device():
    with pytest.raises(Exception):
        load_backend("relay", dict(target="cuda", device_id=31))

コード例 #4

ファイル: test_relay.py プロジェクト: leozhoupeng83/myia

def test_relay_backend_bad_device():
    with pytest.raises(Exception):
        load_backend('relay', dict(target='cuda', device_id=31))

コード例 #5

def test_nnvm_backend_bad_device():
    with pytest.raises(Exception):
        load_backend('nnvm', dict(target="cuda", device_id=31))