Function test_arithm_ops_cpu_gpu

dali/test/python/test_dali_cpu_only.py:1187–1212

()

def test_arithm_ops_cpu_gpu():
    pipe = Pipeline(batch_size=batch_size, num_threads=4, device_id=None)
    data = fn.external_source(source=get_data, layout="HWC")
    processed = [
        data * data.gpu(),
        data + data.gpu(),
        data - data.gpu(),
        data / data.gpu(),
        data // data.gpu(),
        data ** data.gpu(),
        data == data.gpu(),
        data != data.gpu(),
        data < data.gpu(),
        data <= data.gpu(),
        data > data.gpu(),
        data >= data.gpu(),
        data & data.gpu(),
        data | data.gpu(),
        data ^ data.gpu(),
    ]
    pipe.set_outputs(*processed)
    assert_raises(
        RuntimeError,
        pipe.build,
        glob=("The pipeline requires a CUDA-capable GPU but *"),
    )


@attr("pytorch")