Python get_stream示例

示例#1

文件： test_base.py 项目： libocca/occa.py

def test_set_stream():
    host = occa.host()
    stream = host.stream

    assert host == occa.get_device()

    stream2 = host.create_stream()

    assert stream != stream2

    occa.set_stream(stream2)

    assert stream != occa.get_stream()

示例#2

文件： test_base.py 项目： libocca/occa.py

def test_get_stream():
    stream = occa.get_stream()

    assert isinstance(stream, occa.Stream)